Update app.py

app.py

@@ -1,28 +1,3 @@
 import gradio as gr
-import torch
-from diffusers import StableDiffusionPipeline
 
-# Load the model
-model_path = "TheWeirdo69/TestingNSfw"
-pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float32, use_safetensors=True, safety_checker=None)
-pipe.to("cuda" if torch.cuda.is_available() else "cpu")
-
-def generate_image(prompt, height, width):
-    # Generate the image
-    with torch.no_grad():
-        image = pipe(prompt, height=height, width=width, guidance_scale=7.5).images[0]
-    return image
-
-# Define the Gradio interface
-prompt_input = gr.Textbox(lines=2, placeholder="Enter your prompt here...")
-height_input = gr.Slider(minimum=256, maximum=1024, step=8, value=512, label="Height")
-width_input = gr.Slider(minimum=256, maximum=1024, step=8, value=512, label="Width")
-output_image = gr.Image(type="pil")
-
-gr.Interface(
-    fn=generate_image,
-    inputs=[prompt_input, height_input, width_input],
-    outputs=output_image,
-    title="Stable Diffusion Image Generator",
-    description="Generate images with Stable Diffusion model."
-).launch()
+gr.load("models/TheWeirdo69/TestingNSfw").launch()