Spaces:
Runtime error
Runtime error
import gradio as gr | |
import requests | |
import io | |
from PIL import Image | |
import os | |
API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5" | |
HF_token =os.getenv("HF_Token") | |
headers = {"Authorization": f"Bearer {HF_token}"} | |
def pipe(prompt): | |
input_data = {"inputs": prompt} | |
response = requests.post(API_URL, headers=headers, json=input_data) | |
image = Image.open(io.BytesIO(response.content)) | |
return image | |
iface = gr.Interface( | |
pipe, | |
["text"], | |
["image"], | |
allow_flagging="never", | |
examples= [['Astronaut riding a horse'], ['Train is flying in the sky']] | |
) | |
iface.launch() |