Update app.py

app.py

@@ -97,7 +97,7 @@ def test(gpu_id, net, img_list, group_size, img_size):
         #result.convert('L').save('0.png')
         print('done')
         return result
-test('cpu',net,[(torch.rand(224,224,3)*255).numpy().astype(np.uint8) for i in range(5)],5,224)
+test('cpu',net,[(torch.rand(352,352,3)*255).numpy().astype(np.uint8) for i in range(5)],5,224)
 def sepia(img1,img2,img3,img4,img5):
   print('sepia')
   '''ans=[]