Update app.py

app.py

@@ -123,7 +123,8 @@ 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 np.concatenate([img1,img2,img3,img4,img5],axis=1)
+  white=(torch.ones(img1.shape[0],2,3)*255).numpy().astype(np.uint8)
+  return np.concatenate([img1,white,img2,white,img3,white,img4,white,img5],axis=1)
 
 #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")