Spaces:
Runtime error
Runtime error
Commit
·
be1715d
1
Parent(s):
22e8512
updating app
Browse files
app.py
CHANGED
@@ -10,5 +10,6 @@ def predict(img):
|
|
10 |
pred,pred_idx,probs = learn.predict(img)
|
11 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
12 |
|
13 |
-
|
14 |
-
|
|
|
|
10 |
pred,pred_idx,probs = learn.predict(img)
|
11 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
12 |
|
13 |
+
title = "Pokeclassifier"
|
14 |
+
description = "A pokemon classifier trained with fastai."
|
15 |
+
gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), title=title, description=description, outputs=gr.outputs.Label(num_top_classes=3)).launch(share=False)
|