Update app.py

app.py

@@ -4,6 +4,7 @@ from langchain.chains import LLMChain
 from langchain.llms import HuggingFaceHub
 from langchain.prompts import PromptTemplate
 import requests
+import base64
 import streamlit as st
 import json
 
@@ -75,6 +76,7 @@ def generate_image(topic):
   # You can access the image with PIL.Image for example
 
   image = Image.open(io.BytesIO(image_bytes))
+  image.save(f"{topic}.png")
   return image
 
 # Streamlit app