Create app.py

app.py

@@ -0,0 +1,35 @@
+import streamlit as st
+from PIL import Image
+import io
+import requests
+import os 
+HF_TOKEN = os.environ["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
+        
+
+        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+        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()