Update app.py

app.py

@@ -1,5 +1,6 @@
 from transformers import pipeline
 import streamlit as st
+from diffusers import DiffusionPipeline
 
 # Text generation pipeline
 generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
@@ -7,6 +8,9 @@ generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", tru
 # Summarization pipeline
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
+# Image generation pipeline
+image_pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+
 # Input text
 prompt = st.text_input("Enter a prompt to expand on:")
 
@@ -22,3 +26,7 @@ if prompt:
     summarized_text = summary[0]['summary_text']
     st.write("Summarized Text:")
     st.write(summarized_text)
+
+    # Generate an image from the summarized text
+    image = image_pipe(summarized_text).images[0]
+    st.image(image, caption="Generated Image")