Update app.py

app.py

@@ -1,28 +1,7 @@
-
 import gradio as gr
-from diffusers import StableDiffusionPipeline
-import torch
-
-# Load the model
-model_id = "Blane187/ai-hoshino-s1-ponyxl-lora-nochekaise"
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16 if device == "cuda" else torch.float32)
-pipe.to(device)
 
-# Define the inference function
-def generate_image(prompt):
-    with torch.autocast(device):
-        image = pipe(prompt).images[0]
-    return image
+model = "Blane187/ai-hoshino-s1-ponyxl-lora-nochekaise"
 
-# Create Gradio Interface
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
-    outputs=gr.Image(type="pil"),
-    theme="Blane187/fuchsia", 
-    examples=[["ai hoshino, long hair, bangs, purple eyes, purple hair, symbol-shaped pupils,"]]
-)
 
-# Launch the interface
-iface.launch()
+demo = gr.load(model, src="models", theme="Blane187/fuchsia")
+demo.launch()