Spaces:

Ilde
/

test

Sleeping

Ilde commited on Jul 17, 2023

Commit

6fe2844

•

1 Parent(s): ffe36a0

porfabor señor jesus

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ categories = ("Marth", "Sheik")
 def classify_images(img):
     pred, idx, probs = learn.predict(img)
-    return dict(zip(catogories, map(float, probs)))
 image = gr.inputs.Image(shape = (192, 192))
 label = gr.outputs.Label()

 def classify_images(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()