Update app.py

app.py

@@ -30,19 +30,19 @@ if st.button("Process"):
             # Perform text summarization
             summary = summarizer(text_input, max_length=130, min_length=30, do_sample=False)
             summarized_text = summary[0]['summary_text']
-            # Display the summary result
-            st.write("Summarized Text:", summarized_text)
         except Exception as e:
             st.error(f"Error during summarization: {e}")
 
-        try:
-            # Perform question classification on the summarized text
-            results = classifier(summarized_text)[0]
-            # Find the category with the highest score
-            max_score = max(results, key=lambda x: x['score'])
-            st.write("Category:", max_score['label'])
-            st.write("Score:", max_score['score'])
-        except Exception as e:
-            st.error(f"Error during classification: {e}")
+        if summarized_text:
+            try:
+                # Perform question classification on the summarized text
+                results = classifier(summarized_text)[0]
+                # Find the category with the highest score
+                max_score = max(results, key=lambda x: x['score'])
+                st.write("Summarized Text:", summarized_text)
+                st.write("Category:", max_score['label'])
+                st.write("Score:", max_score['score'])
+            except Exception as e:
+                st.error(f"Error during classification: {e}")
     else:
         st.warning("Please enter text to process and ensure both models are loaded.")