Update app.py

app.py

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