Update app

app.py

@@ -45,25 +45,28 @@ if st.button('Summarize'):
                 logits = model(input_values).logits
                 predicted_ids = torch.argmax(logits, dim=-1)
 
-                # Decode the speech
-                transcription = tokenizer.batch_decode(predicted_ids, skip_special_tokens=True)[0]  # Ensure transcription is a single string
+                # Decode the speech to text
+                transcription = tokenizer.batch_decode(predicted_ids, skip_special_tokens=True)[0]
 
                 # Ensure transcription is a string and not empty
                 if isinstance(transcription, str) and transcription.strip():
+                    st.write("Transcription:", transcription)  # Debugging print
+
                     # Show progress
                     st.progress(75)
 
                     # Summarization
                     summarizer = pipeline("summarization")
-                    summary = summarizer(transcription, max_length=130, min_length=30, do_sample=False)
-
-                    # Display the summary
-                    st.success("Done!")
-                    st.write("### Summary:")
-                    st.write(summary[0]['summary_text'])
+                    try:
+                        summary = summarizer(transcription, max_length=130, min_length=30, do_sample=False)
+                        st.success("Done!")
+                        st.write("### Summary:")
+                        st.write(summary[0]['summary_text'])
 
-                    # Final progress
-                    st.progress(100)
+                        # Final progress
+                        st.progress(100)
+                    except Exception as e:
+                        st.error(f"Error in summarization: {e}")  # More specific error message
                 else:
                     st.error("Could not transcribe audio or transcription is empty.")