Update app.py

app.py

@@ -1,18 +1,30 @@
 import gradio as gr
+import requests
+from PIL import Image
+import io
+import os 
 
-model = "Blane187/memcho-s1-ponyxl-lora-nochekaiser"
 
+API_URL = "https://api-inference.huggingface.co/models/Blane187/memcho-s1-ponyxl-lora-nochekaiser"
+API_TOKEN = os.getenv("HF_READ_TOKEN")
 
-examples = [
-    ["memcho, blonde hair, blue eyes, blunt bangs, demon horns, fake horns, horns, short hair,"]
-]
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+def query(inputs):
+    response = requests.post(API_URL, headers=headers, json={"inputs": inputs})
+    image_bytes = response.content
+    image = Image.open(io.BytesIO(image_bytes))
+    return image
+
+with gr.Blocks(theme="Hev832/niceandsimple") as demo:
+    gr.Markdown("# memcho-s1-ponyxl-lora-nochekaiser")
+    
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Enter a prompt", placeholder="Astronaut riding a horse")
+        generate_btn = gr.Button("Generate Image")
+    
+    output_image = gr.Image(label="Generated Image")
+
+    generate_btn.click(fn=query, inputs=prompt_input, outputs=output_image)
 
-demo = gr.load(
-    model, 
-    title="Mem-Cho, PonyXL lora stable diffusion",
-    description="see triger word here https://huggingface.co/Blane187/memcho-s1-ponyxl-lora-nochekaiser",
-    src="models",
-    theme="Blane187/fuchsia",
-    examples=examples,
-)
 demo.launch()