# 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()