Spaces:
Sleeping
Sleeping
Better title
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
#
|
2 |
-
#
|
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="
|
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()
|