Create app.py

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+from PIL import Image
+import torch
+
+# Load model and feature extractor
+model_id = "shahad-alh/arabichar-finetuned-v2"
+model = AutoModelForImageClassification.from_pretrained(model_id)
+extractor = AutoFeatureExtractor.from_pretrained(model_id)
+
+# Prediction function
+def predict(img: Image.Image):
+    inputs = extractor(images=img, return_tensors="pt")
+    with torch.no_grad():
+        outputs = model(**inputs)
+        predicted = outputs.logits.argmax(-1).item()
+        label = model.config.id2label[str(predicted)]
+    return label
+
+gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil", label="Upload Arabic Letter"),
+    outputs=gr.Textbox(label="Prediction"),
+    allow_flagging="never",
+    allow_api=True,
+    title="Arabic Character Classifier"
+).launch()