File size: 628 Bytes
078ce08 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | import os
import requests
import io
from PIL import Image
API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
def generate(text_prompt):
api_key = os.environ.get("HF_API_KEY", "")
if not api_key:
raise ValueError("HF_API_KEY environment variable is not set.")
response = requests.post(
API_URL,
headers={"Authorization": f"Bearer {api_key}"},
json={"inputs": text_prompt},
)
response.raise_for_status()
return Image.open(io.BytesIO(response.content))
"""
text = input("Your sentence here:")
image = generate(text)
image.show()
""" |