Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Load model
+
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float32)
 pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Disable safety checker (optional)
+
 pipe.safety_checker = lambda images, **kwargs: (images, False)
 
-# Define inference function
+
 def infer(prompt, guidance_scale, num_inference_steps):
     with torch.no_grad():
         try:
@@ -18,7 +18,7 @@ def infer(prompt, guidance_scale, num_inference_steps):
         except Exception as e:
             return f"Error: {str(e)}"
 
-# Gradio UI
+
 with gr.Blocks() as demo:
     gr.Markdown("🚀 Hyper-Sketch")