Upload folder using huggingface_hub

README.md

@@ -1,12 +1,2 @@
----
-title: Vit Image Classification
-emoji: 🐠
-colorFrom: green
-colorTo: gray
-sdk: gradio
-sdk_version: 5.45.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# ViT Image Classification
+Uses google/vit-base-patch16-224.

app.py

@@ -0,0 +1,21 @@
+
+import gradio as gr
+from transformers import pipeline
+
+clf = pipeline("image-classification",
+               model="google/vit-base-patch16-224",
+               device_map="auto")
+
+def predict(img):
+    out = clf(img)[:5]
+    return {o['label']: float(o['score']) for o in out}
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=5),
+    title="Image Classification (ViT)"
+)
+
+if __name__ == "__main__":
+    demo.launch(share=True)  # share=True gives you a temporary Colab link

requirements.txt

@@ -0,0 +1,6 @@
+gradio>=4.44.0
+transformers>=4.43.0
+accelerate>=0.33.0
+sentencepiece
+safetensors
+torch