ImageGenerator / app.py
AjithBharadwaj's picture
Update app.py
4878b34 verified
import streamlit as st
from PIL import Image
import io
import requests
import os
HF_TOKEN = os.environ["HF_TOKEN"]
API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
headers = {"Authorization": f"Bearer {HF_TOKEN}"}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
def main():
st.title(":male-astronaut: Image Generator App :male-astronaut:")
# Text input box for the user
text_input = st.text_input("Enter your Imagination :")
# Button to trigger image generation
if st.button("Generate Image"):
# Call image_generator function
image_bytes = query({
"inputs": text_input,
})
image = Image.open(io.BytesIO(image_bytes))
# Display the generated image
st.image(image, caption='Generated Image', use_column_width=True)
if __name__ == "__main__":
main()