turtlegraphics commited on
Commit
60c3596
·
verified ·
1 Parent(s): 384ba32

Better title

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  #
2
- # gradio demo
3
  #
4
  import gradio as gr
5
  from transformers import AutoImageProcessor, AutoModelForImageClassification
@@ -14,6 +14,6 @@ def classify(image):
14
  predicted_class_idx = logits.argmax(-1).item()
15
  return model.config.id2label[predicted_class_idx]
16
 
17
- demo = gr.Interface(fn=classify, inputs="image", outputs="text", title="Sandbox")
18
 
19
  demo.launch()
 
1
  #
2
+ # Image Recognition with MobileNetv2
3
  #
4
  import gradio as gr
5
  from transformers import AutoImageProcessor, AutoModelForImageClassification
 
14
  predicted_class_idx = logits.argmax(-1).item()
15
  return model.config.id2label[predicted_class_idx]
16
 
17
+ demo = gr.Interface(fn=classify, inputs="image", outputs="text", title="Image Recognition With MobileNetv2")
18
 
19
  demo.launch()