Update app.py

app.py

@@ -120,9 +120,9 @@ def sepia(img1,img2,img3,img4,img5):
   result_list=test(device,net,img_list,5,224)
   #result_list=[result_list[i].resize((w_list[i], h_list[i]), Image.BILINEAR) for i in range(5)]
   img1,img2,img3,img4,img5=result_list#test('cpu',net,img_list,5,224)
-  return img1,img2,img3,img4,img5
+  return np.concatenate([img1,img2,img3,img4,img5],axis=0)
 
 #gr.Image(shape=(224, 2))
-demo = gr.Interface(sepia, inputs=["image","image","image","image","image"], outputs=["image","image","image","image","image"])#gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
-
+#demo = gr.Interface(sepia, inputs=["image","image","image","image","image"], outputs=["image","image","image","image","image"])#gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
+demo = gr.Interface(sepia, inputs=["image","image","image","image","image"], outputs=["image"])
 demo.launch(debug=True)