Update app.py

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 from diffusers import DiffusionPipeline
 
 #Text generation pipeline
-generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
+#generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
 
 #Summarization pipeline
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
@@ -18,13 +18,13 @@ prompt = st.text_input("Enter a prompt to expand on:")
 #Check if text has been entered into the prompt
 if prompt:
     # Generate text based on the prompt
-    generated_text = generator(prompt, max_length=300, num_return_sequences=1)
-    detailed_text = generated_text[0]['generated_text']
-    st.write("Generated Text:")
-    st.write(detailed_text)
+    #generated_text = generator(prompt, max_length=300, num_return_sequences=1)
+    #detailed_text = generated_text[0]['generated_text']
+    #st.write("Generated Text:")
+    #st.write(detailed_text)
 
     # Summarize the generated text
-    summary = summarizer(detailed_text, max_length=130, min_length=30, do_sample=False)
+    summary = summarizer(prompt, max_length=130, min_length=30, do_sample=False)
     summarized_text = summary[0]['summary_text']
     st.write("Summarized Text:")
     st.write(summarized_text)