Spaces:
Runtime error
Runtime error
File size: 617 Bytes
7c7c528 1654adb c9591a9 cb0f5a4 c9591a9 cb0f5a4 c9591a9 cb0f5a4 4a73852 e1d7322 cb0f5a4 134b12c 0f8ee2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import os
os.system("pip install baseten")
os.system("pip install python-dotenv")
import gradio as gr
import baseten
from baseten.models import StableDiffusionPipeline
from dotenv import load_dotenv
load_dotenv()
BASETEN_API_KEY = os.getenv("BASETEN_API_KEY")
baseten.login(BASETEN_API_KEY)
model = StableDiffusionPipeline(
"runwayml/stable-diffusion-v1-5",
vae="stabilityai/sd-vae-ft-mse",
scheduler="pndm"
)
def t2i(text):
image, url = model(text, seed=2)
return image
iface = gr.Interface(fn=t2i, inputs="text", outputs=[gr.Image(label="Generated Image")]).launch()
iface.launch(share=True) |