Spaces:

Plachta
/

GraspAnything

Running

Plachta commited on Feb 21

Commit

e7d17af

•

1 Parent(s): a1b7ca6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -106,7 +106,7 @@ def predict(input, topk):
     recovered_img = recovered_img[:t_orig_size[0], :t_orig_size[1]]
     # resize to original size
-    recovered_img = cv2.resize(recovered_img, (orig_size[1]， orig_size[0]))
     return recovered_img
 if __name__ == "__main__":

     recovered_img = recovered_img[:t_orig_size[0], :t_orig_size[1]]
     # resize to original size
+    recovered_img = cv2.resize(recovered_img, (orig_size[1], orig_size[0]))
     return recovered_img
 if __name__ == "__main__":