Spaces:

osanseviero
/

draw_to_search

Runtime error

osanseviero commited on Oct 12, 2021

Commit

c48dde4

1 Parent(s): 52fb80a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def download_img(path):
         handler.write(img_data)
     return local_path
-def predict(query):
     x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.
     with torch.no_grad():
         out = class_model(x)

         handler.write(img_data)
     return local_path
+def predict(im):
     x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.
     with torch.no_grad():
         out = class_model(x)