mostafa add notifction API

app.py

@@ -0,0 +1,38 @@
+import gradio as gr
+from PIL import Image
+import numpy as np
+
+from transformers import AutoFeatureExtractor,AutoModelForImageClassification,pipeline
+
+model_id = "smartgmin/Entrnal_5class_agumm_last_newV7_model" 
+ 
+
+model2 = AutoModelForImageClassification.from_pretrained(model_id, from_tf=True) 
+ 
+
+feature_extractor = AutoFeatureExtractor.from_pretrained(model_id)
+
+
+clsss = pipeline('image-classification',model=model2,feature_extractor=feature_extractor)
+
+def predict(image):
+  yl = clsss(image)
+  max_item = max(yl, key=lambda x: x['score'])
+  nn = "{:f}".format(max_item['score'])
+  dd = max_item['label']
+  nn,dd
+  return f"dissesse: {dd} , score : {nn}"
+
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.inputs.Image(type="pil", label="ارفع صورتك"),
+    outputs="text",
+    title="نموذج ViT لتصنيف الصور",
+    description="قم برفع صورة للحصول على تصنيف باستخدام نموذج ViT المدرب."
+)
+
+
+iface.launch()
+
+

requirements.txt

@@ -0,0 +1,8 @@
+gradio
+PIL
+numpy
+transformers==4.44.2
+tensorflow==2.17.0
+torch==2.4.1 
+torchvision==0.19.1 
+torchaudio==2.4.1