Spaces:

uhlarlar
/

basicPetClassifier

Sleeping

uhlarlar commited on Feb 20, 2023

Commit

8a20b98

1 Parent(s): 63b90bf

fixed another typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ categories = learn.dls.vocab
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
-    return dict(zip(categories, (map(float, probs))))
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label()

 def classify_image(img):
     pred, idx, probs = learn.predict(img)
+    return dict(zip(categories, map(float, probs)))
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label()