Update app.py

app.py

@@ -1,19 +1,21 @@
+#Import dependencies
 from transformers import pipeline
 import streamlit as st
 from diffusers import DiffusionPipeline
 
-# Text generation pipeline
+#Text generation pipeline
 generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
 
-# Summarization pipeline
+#Summarization pipeline
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 # Image generation pipeline
-image_pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+#image_pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
 
 # Input text
 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)
@@ -28,5 +30,5 @@ if prompt:
     st.write(summarized_text)
 
     # Generate an image from the summarized text
-    image = image_pipe(summarized_text).images[0]
-    st.image(image, caption="Generated Image")
+    #image = image_pipe(summarized_text).images[0]
+    #st.image(image, caption="Generated Image")