Spaces:

raees
/

Riot-Detector

Runtime error

raees commited on Sep 22, 2022

Commit

0ccdef4

•

1 Parent(s): 36b4b34

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ learner = load_learner('model.pkl')
 categories = ("Celebration", "Riot")
 def classify_image(image):
-    pred, idx, probs = learner.predict(img)
     return dict(zip(categories, map(float, probs)))
 image = gr.inputs.Image(shape=(192, 192))

 categories = ("Celebration", "Riot")
 def classify_image(image):
+    pred, idx, probs = learner.predict(image)
     return dict(zip(categories, map(float, probs)))
 image = gr.inputs.Image(shape=(192, 192))