Update app.py

app.py

@@ -101,8 +101,8 @@ def test(gpu_id, net, img_list, group_size, img_size):
         print('done')
         return result
 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
+'''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=[]