Upload page4.py

page4.py

@@ -0,0 +1,27 @@
+import streamlit as st
+import requests
+import time
+import os
+import random
+from PIL import Image
+from io import BytesIO
+import io
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+def query_model(text_input):
+        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+        payload = {"inputs": text_input, "num_inference_steps": 18, "guidance_scale": 4, "seed": random.randint(1, 9999999), "width": 1024, "height": 1024, "negative_prompt": "blurry, ugly, deformed, bad anatomy"}
+        response = requests.post(API_URL, headers=headers, json=payload)
+        return response.content
+
+def sdxl():
+    st.write("ForgeImage")
+    text_input = st.text_input("Enter your prompt:", "Astronaut riding a horse")
+
+    if st.button("Create"):   
+        image_bytes = query_model(text_input)
+        generated_image = Image.open(io.BytesIO(image_bytes))
+
+    st.image(generated_image, use_column_width=True)