Spaces:
Runtime error
Runtime error
File size: 663 Bytes
31d5745 eab793d 31d5745 eab793d 7dfb744 31d5745 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 27 28 29 30 |
# text to Image on Gradio UI
import torch
from diffusers import DiffusionPipeline
import gradio as gr
# Provide the file path to your locally stored model file
model_file_path = "/model/file/path/to/model/directory/"
# Provide the url to model id
url = "runwayml/stable-diffusion-v1-5"
# Load the safetensor model
pipe = DiffusionPipeline.from_pretrained(url, torch_dtype=torch.float16)
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()
|