Hugging Face's logo

Spaces:
bareethul
/
stop-sign-predictor
Sleeping

App Files Community
bareethul commited on
Commit
847ccdc
·
verified ·
1 Parent(s): 617590f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def do_predict(pil_img: PIL.Image.Image):
27
  # Save to tmp for AutoGluon input
28
  pil_resized.save(TMP_PATH)
29
 
30
- df = pd.DataFrame({"image": [str(img_path)]})
31
 
32
  # Predict probabilities
33
  proba_df = PREDICTOR.predict_proba(df)
 
27
  # Save to tmp for AutoGluon input
28
  pil_resized.save(TMP_PATH)
29
 
30
+ df = pd.DataFrame({"image": [str(TMP_PATH)]})
31
 
32
  # Predict probabilities
33
  proba_df = PREDICTOR.predict_proba(df)