Spaces:

yasirapunsith
/

vpr_v3

Paused

yasirapunsith commited on Nov 3, 2023

Commit

9e436c7

1 Parent(s): 91b60f7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -493,8 +493,8 @@ model = torch.load("model.pt")
 def predict(image):
     """Generates a prediction for the given image."""
     # image = FileData(image)
-    image = components.Image(image)
-    # image = Image.fromarray(np.uint8(image))
     model.eval()

 def predict(image):
     """Generates a prediction for the given image."""
     # image = FileData(image)
+    # image = components.Image(image)
+    image = Image.fromarray(np.uint8(image))
     model.eval()