Update app.py

app.py

@@ -1,13 +1,13 @@
-def diffusion(text,num_inference_steps,guidance_scale):
-    from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
-
-    model_id = "stabilityai/stable-diffusion-2-1"
+import gradio as gr
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
-    # Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-    pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-    pipe = pipe.to("cuda")
+model_id = "stabilityai/stable-diffusion-2-1"
 
+# Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+pipe = pipe.to("cuda")
+def diffusion(text,num_inference_steps,guidance_scale):
     prompt = text
     image = pipe(prompt,guidance_scale=guidance_scale, num_inference_steps=num_inference_steps).images[0]
     return image