Spaces:
Runtime error
Runtime error
File size: 702 Bytes
31d5745 eab793d 31d5745 eab793d 7dfb744 31d5745 41108f2 ceee367 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 31 |
# 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)
#to use metal performance shader for gpu inference optimization 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()
|