Spaces:

Frantz103
/

cat_dogs

Sleeping

Frantz103 commited on Apr 23, 2023

Commit

8869ce8

•

1 Parent(s): fc2a335

update labels

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ def is_cat(x): return x[0].isupper()
 learn_inf = load_learner('DogCat.pkl')
 labels = learn_inf.dls.vocab
-labels = ['Cat', 'dog']
 def predict(img):
     img = PILImage.create(img)
@@ -15,4 +15,4 @@ def predict(img):
 title = "Is this image a cat?"
 iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512,512)), outputs=gr.outputs.Label(num_top_classes=2), title=title)
-iface.launch()

 learn_inf = load_learner('DogCat.pkl')
 labels = learn_inf.dls.vocab
+labels = ['Cat', 'Dog']
 def predict(img):
     img = PILImage.create(img)
 title = "Is this image a cat?"
 iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512,512)), outputs=gr.outputs.Label(num_top_classes=2), title=title)
+iface.launch(share=True)