update

app.py

@@ -22,10 +22,10 @@ def inference_mask1(prompt,
     a = json.loads(r.text)
     res = []
     for i in range(len(a)):
-        out = Image.open(io.BytesIO(base64.b64decode(a[i])))
-        out = out.resize((224, 224))
-        res.append(np.uint8(np.array(out)))
-        #res.append(np.uint8(np.array(Image.open(io.BytesIO(base64.b64decode(a[i]))))))
+        #out = Image.open(io.BytesIO(base64.b64decode(a[i])))
+        #out = out.resize((224, 224))
+        #res.append(np.uint8(np.array(out)))
+        res.append(np.uint8(np.array(Image.open(io.BytesIO(base64.b64decode(a[i]))))))
     return res
 
 def resizeImg(img):
@@ -58,7 +58,8 @@ examples = [
 
 demo_mask = gr.Interface(fn=inference_mask1, 
                    inputs=[gr.ImageMask(label="prompt (提示图)"), gr.Image(label="img1 (测试图1)"), gr.Image(label="img2 (测试图2)")], 
-                    outputs=[gr.Image(label="output1 (输出图1)"), gr.Image(label="output2 (输出图2)")],
+                    outputs=[gr.Image(label="output1 (输出图1)", shape=(448, 448)), gr.Image(label="output2 (输出图2)"), shape=(448, 448)],
+                    #outputs=[gr.Image(label="output1 (输出图1)"), gr.Image(label="output2 (输出图2)")],
                     #outputs=gr.Gallery(label="outputs (输出图)"),
                     examples=examples,
                     #title="SegGPT for Any Segmentation<br>(Painter Inside)",