Spaces:

muellerzr
/

fastai-pets-resnet-34

Runtime error

muellerzr commited on Nov 22, 2021

Commit

30c9eef

1 Parent(s): 21be48b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,5 +21,5 @@ def predict(img):
     clas, idx, probs = learn.predict(img)
     return "Dog" if clas == "False" else "Cat"
-iface = gr.Interface(fn=predict, inputs="image", outputs="text")
 iface.launch()

     clas, idx, probs = learn.predict(img)
     return "Dog" if clas == "False" else "Cat"
+iface = gr.Interface(fn=predict, inputs="image", outputs="label")
 iface.launch()