Update app.py

app.py

@@ -14,7 +14,7 @@ try:
     labels = response.json()
     if isinstance(labels, dict):
         labels = list(labels.values())
-except requests.exceptions.RequestException as e:
+except Exception as e:
     print("فشل تحميل labels.json:", e)
     labels = None
 
@@ -23,20 +23,24 @@ pipe = pipeline("text-classification", model=model_name)
 
 # دالة التنبؤ
 def predict(text):
-    result = pipe(text)[0]
-    label_id = int(result['label'].replace("LABEL_", ""))  # تحويل LABEL_3 إلى 3
-    label = labels[label_id] if labels and label_id < len(labels) else result['label']
-    confidence = round(result['score'], 3)
+    try:
+        result = pipe(text)[0]
+        label_id = int(result['label'].replace("LABEL_", ""))  # تحويل LABEL_3 إلى 3
+        label = labels[label_id] if labels and label_id < len(labels) else result['label']
+        confidence = round(result['score'], 3)
 
-    json_output = {
-        "prediction": label,
-        "confidence": confidence
-    }
+        json_output = {
+            "prediction": label,
+            "confidence": confidence
+        }
 
-    return label, json.dumps(json_output, indent=4, ensure_ascii=False)
+        return label, json.dumps(json_output, indent=4, ensure_ascii=False)
+    
+    except Exception as e:
+        return "خطأ في المعالجة", json.dumps({"error": str(e)}, indent=4, ensure_ascii=False)
 
-# Gradio
-gr.Interface(
+# Gradio Interface
+demo = gr.Interface(
     fn=predict,
     inputs=gr.Textbox(lines=3, placeholder="اكتب جملة باللهجة المصرية..."),
     outputs=[
@@ -45,4 +49,7 @@ gr.Interface(
     ],
     title="تصنيف النصوص باللهجة المصرية",
     description="نموذج يصنف النصوص إلى: حياد، إساءة، عنصرية، تمييز ديني، إعلانات، إلخ."
-).launch(share=True)
+)
+
+if __name__ == "__main__":
+    demo.launch()