Spaces:
Runtime error
Runtime error
import requests | |
import gradio as gr | |
import io | |
from PIL import Image | |
import hashlib | |
API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0" | |
headers = {"Authorization": {HG} } | |
def query(payload): | |
response = requests.post(API_URL, headers=headers, json=payload) | |
return response.content | |
def generate_image(prompt): | |
# Add a unique identifier to the prompt | |
prompt_with_identifier = prompt + str(hashlib.md5(prompt.encode()).hexdigest()) | |
image_bytes = query({ | |
"inputs": prompt_with_identifier, | |
}) | |
image = Image.open(io.BytesIO(image_bytes)) | |
return image | |
iface = gr.Interface(generate_image, | |
inputs="text", | |
outputs="image", | |
title="Generate Image from Text", | |
description="Enter a text prompt to generate a new image each time.", | |
allow_flagging=False) | |
iface.launch() | |