mostafa add notifction API6

app.py

@@ -1,21 +1,18 @@
 import gradio as gr
-from PIL import Image
+from PIL import Image, UnidentifiedImageError
 import numpy as np
 import requests
 from io import BytesIO
 
-from transformers import AutoFeatureExtractor,AutoModelForImageClassification,pipeline
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification, pipeline
 
-model_id = "smartgmin/Entrnal_5class_agumm_last_newV7_model" 
- 
-
-model2 = AutoModelForImageClassification.from_pretrained(model_id, from_tf=True) 
- 
+# معرف النموذج الخاص بك على Hugging Face
+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)
+clsss = pipeline('image-classification', model=model2, feature_extractor=feature_extractor)
 
 def predict(image_url):
     try:
@@ -31,6 +28,7 @@ def predict(image_url):
         # تحميل الصورة من الرابط
         image = Image.open(BytesIO(response.content)).convert("RGB")
 
+        # تمرير الصورة إلى النموذج
         yl = clsss(image)
         max_item = max(yl, key=lambda x: x['score'])
         nn = "{:.2f}".format(max_item['score'])  # تنسيق الدقة ليكون مقروءًا بشكل أفضل
@@ -44,10 +42,10 @@ def predict(image_url):
     except Exception as e:
         return f"حدث خطأ أثناء معالجة الصورة: {e}"
 
-# إنشاء واجهة Gradio
+# إنشاء واجهة Gradio باستخدام المكونات الجديدة
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.inputs.Textbox(lines=2, placeholder="أدخل رابط الصورة هنا...", label="رابط الصورة"),
+    inputs=gr.Textbox(lines=2, placeholder="أدخل رابط الصورة هنا...", label="رابط الصورة"),
     outputs="text",
     title="نموذج ViT لتصنيف الصور",
     description="أدخل رابط صورة للحصول على تصنيف باستخدام نموذج ViT المدرب."
@@ -55,4 +53,3 @@ iface = gr.Interface(
 
 # تشغيل الواجهة
 iface.launch()
-