software / AIMAGE.py
shethjenil's picture
Update AIMAGE.py
f49cc2f verified
raw
history blame
510 Bytes
from requests import post
from io import BytesIO
from PIL.Image import open as open_image
from tkinter.simpledialog import askstring
while True:
text = askstring("IMAGE PROMPT",prompt="What is your prompt?")
if text and text.strip() != "":
hfapi = "<KEY>"
open_image(BytesIO(post("https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0", headers={"Authorization": f"Bearer {hfapi}"}, json={"inputs": text.strip()}).content)).show()
else:
break