nice interface

README.md

@@ -1,8 +1,8 @@
 ---
 title: Fastai
-emoji: 📉
-colorFrom: pink
-colorTo: yellow
+emoji: 🐸 
+colorFrom: yellow
+colorTo: green
 sdk: gradio
 sdk_version: 3.1.1
 app_file: app.py

app.py

@@ -3,11 +3,17 @@ import gradio as gr
 
 learn = load_learner('export.pkl')
 
-labels = learn.dls.vocab
+labels = ["This is Battletoads!", "Not Battletoads"]
 
 def predict(img):
     img = PILImage.create(img)
     pred,pred_idx,probs = learn.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
-gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)),examples=["battle.jpg"],outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)
+title = "Is it Battletoads game?"
+description = '<img src="https://huggingface.co/spaces/HuggingDavid/battletoads/resolve/main/pic.jpg" alt="rick" width="200"/>'
+examples = ['battle.jpg']
+interpretation='default'
+enable_queue=True
+
+gr.Interface(fn=predict,inputs=gr.inputs.Image(shape=(512, 512)),outputs=gr.outputs.Label(num_top_classes=3),title=title,examples=examples,interpretation=interpretation,enable_queue=enable_queue).launch()

requirements.txt

@@ -1 +1,2 @@
-fastai
+fastai
+gradio