Update app.py

app.py

@@ -1,26 +1,27 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 
-# יצירת מודל לניתוח רגשות
-sentiment_analysis = pipeline(
-    "sentiment-analysis",
-    model="avichr/heBERT_sentiment_analysis",
-    tokenizer="avichr/heBERT_sentiment_analysis",
-    return_all_scores = True
-)
-
-# פונקציה לניתוח הטקסט
-def analyze_sentiment(text):
-    results = sentiment_analysis(text)[0]
-    return {r['label']: f"{r['score']:.2%}" for r in results}
+def analyze_text(text):
+    try:
+        tokenizer = AutoTokenizer.from_pretrained("avichr/heBERT_sentiment_analysis")
+        model = AutoModelForSequenceClassification.from_pretrained("avichr/heBERT_sentiment_analysis")
+        
+        classifier = pipeline(
+            "sentiment-analysis",
+            model=model,
+            tokenizer=tokenizer
+        )
+        
+        result = classifier(text)
+        return str(result)
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-# יצירת הממשק
-iface = gr.Interface(
-    fn=analyze_sentiment,
-    inputs="text",
-    outputs="label",
-    title="ניתוח רגשות בעברית",
-    description="הכנס טקסט בעברית וקבל ניתוח רגשות"
+interface = gr.Interface(
+    fn=analyze_text,
+    inputs=gr.Textbox(label="הכנס טקסט בעברית"),
+    outputs=gr.Textbox(label="תוצאה"),
+    title="ניתוח רגשות בעברית"
 )
 
-iface.launch()
+interface.launch()