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