Update app.py

app.py

@@ -17,7 +17,7 @@ if st.button("Generate Image") and text:
     text_features = clip_processor(text, return_tensors="pt", padding=True)
 
     # Load an example image from the web (replace this with your image loading logic)
-    example_image_url = "https://example.com/your-image.jpg"
+    example_image_url = "https://source.unsplash.com/random"
     example_image_response = requests.get(example_image_url)
     example_image = Image.open(BytesIO(example_image_response.content))
 
@@ -34,7 +34,10 @@ if st.button("Generate Image") and text:
 
     # For visualization, you can convert the image representation back to an image
     image_array = image_representation.squeeze().cpu().numpy()
-    image = Image.fromarray((image_array * 255).astype('uint8'))
+    generated_image = Image.fromarray((image_array * 255).astype('uint8'))
 
     # Display the generated image
-    st.image(image, caption="Generated Image", use_column_width=True)
+    st.image(generated_image, caption="Generated Image", use_column_width=True)
+
+    # Save the generated image to a file (change the file path as needed)
+    generated_image.save("generated_image.jpg")