Update app.py

app.py

@@ -1,7 +1,22 @@
 import gradio as gr
+import baseten
+from baseten.models import StableDiffusionPipeline
+import os
+from dotenv import load_dotenv
+load_dotenv()
 
-def greet(name):
-    return "Hello " + name + "!!"
+BASETEN_API_KEY = os.getenv("BASETEN_API_KEY")
+baseten.login(BASETEN_API_KEY)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+model = StableDiffusionPipeline(
+    "runwayml/stable-diffusion-v1-5",
+    vae="stabilityai/sd-vae-ft-mse",
+    scheduler="pndm"
+)
+
+def t2i(text):
+    image, url = model(text)
+    return url
+
+iface = gr.Interface(fn=t2i, inputs="text", outputs="image").launch()
 iface.launch()