Spaces:

AItool
/

is_virtualstaging

Runtime error

AItool commited on Mar 25, 2024

Commit

c93f364

verified ·

1 Parent(s): 3099663

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ learn = load_learner('model.pkl')
 categories =('Virtual Staging','Real')
 def classify_image(img):
-    pred,idx,probs = learn.predict(im)
     return dict(zip(categories,map(float,probs)))
 #*** We have to cast to float above because KAGGLE does not return number on the answer it returns tensors, and Gradio does not deal with numpy so we have to cast to float

 categories =('Virtual Staging','Real')
 def classify_image(img):
+    pred,idx,probs = learn.predict(img)
     return dict(zip(categories,map(float,probs)))
 #*** We have to cast to float above because KAGGLE does not return number on the answer it returns tensors, and Gradio does not deal with numpy so we have to cast to float