Update app.py

app.py

@@ -9,7 +9,6 @@ load_dotenv()
 
 token = os.environ.get("HF_DIARIZATION_TOKEN")
 
-
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 
@@ -44,7 +43,7 @@ if st.button("Transcribir y Diarizar"):
                 speakers = set()
                 for segment, _, speaker in diarization.itertracks(yield_label=True):
                     diarization_segments.append((segment.start, segment.end, speaker))
-                    diarization_text += f"{segment.start:.2f} - {segment.end:.2f}: Speaker {speaker}\n"
+                    diarization_text += f"{segment.start:.2f} - {segment.end:.2f}: {speaker}\n"
                     speakers.add(speaker)
 
                 num_speakers = len(speakers)
@@ -119,7 +118,7 @@ if st.button("Transcribir y Diarizar"):
                         full_text = full_text[:max_tokens]
 
                     # Usar el modelo de Hugging Face para resumir
-                    summary = summarizer(full_text, max_length=150, do_sample=False)[0]['summary_text']
+                    summary = summarizer(full_text, max_length=150, min_length=30, do_sample=False)[0]['summary_text']
 
                     st.text_area(f"Resumen del Hablante {speaker}", summary, height=200)
             except Exception as e: