Update app.py

app.py

@@ -97,7 +97,9 @@ 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(352,352,3)*255).numpy().astype(np.uint8) for i in range(5)],5,224)
+res=test('cpu',net,[(torch.rand(352,352,3)*255).numpy().astype(np.uint8) for i in range(5)],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):
   print('sepia')
   '''ans=[]