Spaces:
Runtime error
Runtime error
File size: 509 Bytes
eab793d 7dfb744 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from diffusers import StableDiffusionPipeline
import torch
# Provide the url to model id
url = "runwayml/stable-diffusion-v1-5"
# Load the safetensor model
pipe = DiffusionPipeline.from_pretrained(model_file_path, use_safetensors=True )
pipe = pipe.to("mps")
def predict(text):
# Ensure pipe(text) returns the correct output format
generated_image = pipe(text).images[0]
return generated_image
iface = gr.Interface(
fn=predict,
inputs='text',
outputs='image',
)
iface.launch()
|