Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ def predict_image(img):
|
|
14 |
return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(learn.dls.vocab))}
|
15 |
|
16 |
# Create a Gradio interface
|
17 |
-
gr.Interface(fn=
|
18 |
|
19 |
if __name__ == "__main__":
|
20 |
interface.launch()
|
|
|
14 |
return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(learn.dls.vocab))}
|
15 |
|
16 |
# Create a Gradio interface
|
17 |
+
gr.Interface(fn=predict_image,inputs=gr.Image(type= "pil"),outputs=gr.Label())
|
18 |
|
19 |
if __name__ == "__main__":
|
20 |
interface.launch()
|