Update app.py

app.py

@@ -73,9 +73,9 @@ def run_wilow_model(image, conf, IoU_threshold=0.5):
         for batch in dataloader: 
             batch = recursive_to(batch, device)
     
-    #         with torch.no_grad():
-    #             out = model(batch) 
-                
+            with torch.no_grad():
+                out = model(batch) 
+            print(out['pred_vertices'])
     #         multiplier    = (2*batch['right']-1)
     #         pred_cam      = out['pred_cam']
     #         pred_cam[:,1] = multiplier*pred_cam[:,1]