Update app.py

app.py

@@ -1,13 +1,21 @@
 import os
 os.system("pip install requests")
+os.system("pip install pillow")
 
 import requests
 import gradio as gr
+from PIL import Image
 
 def t2i(text):
     prompt_url = f"https://aadarsh-text-to-image-mymvubi2mq-el.a.run.app/t2i/{text}"
-    url = requests.get(prompt_url).json()["url"]
-    return url
+    response = requests.get(prompt_url).json()["url"]
+    if response.status_code:
+        fp = open('image.png', 'wb')
+        fp.write(response.content)
+        fp.close()
+
+    image = Image.open('image.png')
+    return image
 
 iface = gr.Interface(fn=t2i, inputs="text", outputs=[gr.Image(label="Generated Image")]).launch()
 iface.launch(share=True)