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