Update app.py

app.py

@@ -5,16 +5,35 @@ from huggingface_hub import from_pretrained_keras
 import numpy as np
 from keras.preprocessing.sequence import pad_sequences
 from keras.datasets import imdb
+import logging
 
-# טעינת המודל מ-Hugging Face Hub
+# הגדרת לוגר בסיסי
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# הגדרת המודל כמשתנה גלובלי
+global model
+model = None
+
+# פונקציה לטעינת המודל
+def load_model():
+    global model
+    try:
+        logger.info("התחלת טעינת המודל...")
+        model = from_pretrained_keras("GiladtheFixer/Sentiment_Analysis")
+        logger.info("המודל נטען בהצלחה!")
+        return True
+    except Exception as e:
+        logger.error(f"שגיאה בטעינת המודל: {e}")
+        return False
+
+# טעינת אינדקס המילים
 try:
-    model = from_pretrained_keras("GiladtheFixer/Sentiment_Analysis")
-    print("Model loaded successfully!")
+    word_index = imdb.get_word_index()
+    logger.info("אינדקס המילים נטען בהצלחה")
 except Exception as e:
-    print(f"Error loading model: {e}")
-
-# קבלת אינדקס המילים של IMDB
-word_index = imdb.get_word_index()
+    logger.error(f"שגיאה בטעינת אינדקס המילים: {e}")
+    raise
 
 def preprocess_text(text):
     # המרה למילים
@@ -32,6 +51,13 @@ def preprocess_text(text):
     return vector
 
 def predict_sentiment(text):
+    global model
+    
+    # בדיקה אם המודל טעון
+    if model is None:
+        if not load_model():
+            return {"Error": "Failed to load model"}
+    
     try:
         # עיבוד הטקסט
         processed_text = preprocess_text(text)
@@ -47,9 +73,8 @@ def predict_sentiment(text):
             "Confidence": f"{confidence:.2%}"
         }
     except Exception as e:
-        return {
-            "Error": str(e)
-        }
+        logger.error(f"שגיאה בביצוע החיזוי: {e}")
+        return {"Error": str(e)}
 
 # יצירת ממשק Gradio
 iface = gr.Interface(
@@ -68,6 +93,9 @@ iface = gr.Interface(
     theme=gr.themes.Soft()
 )
 
-# הפעלת הממשק
+# טעינת המודל בהפעלה הראשונית
 if __name__ == "__main__":
-    iface.launch(share=True)  # שנה ל-share=False אם אתה לא רוצה לייצר קישור ציבורי
+    if load_model():
+        iface.launch(share=True)
+    else:
+        logger.error("לא ניתן להפעיל את הממשק - המודל לא נטען בהצלחה")