Update app.py

app.py

@@ -8,8 +8,8 @@ from transformers import pipeline
 # הגדרת המודל לתמלול
 model = WhisperModel("ivrit-ai/faster-whisper-v2-d4")
 
-# הגדרת pipeline לסיכום
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+# הגדרת pipeline לסיכום עם מודל מותאם לעברית
+summarizer = pipeline("summarization", model="avichr/heBART")
 
 def transcribe_and_summarize(file_path):
     try:
@@ -23,10 +23,9 @@ def transcribe_and_summarize(file_path):
         segments, _ = model.transcribe(audio_file, language="he")
         transcript = " ".join([segment.text for segment in segments])
 
-        # סיכום התמלול
-        
-        summary = summarizer(transcript, max_length=50, min_length=25, do_sample=False)[0]["summary_text"]
-        prompt_text = f"סכם את התמלול הבא כשיעור אקדמי בעברית:\n{transcript}"
+        # סיכום התמלול עם מודל בעברית
+        summary = summarizer(transcript)[0]["summary_text"]
+
         # מחיקת קובץ האודיו במידת הצורך (אם היה וידאו)
         if audio_file != file_path:
             os.remove(audio_file)