Update app.py

app.py

@@ -11,9 +11,11 @@ pipeline = DiffusionPipeline.from_pretrained(
 )
 pipeline.unet.load_attn_procs("gr33nr1ng3r/Mukh-Oboyob")
 
+
 def diffusion(text,num_inference_steps,guidance_scale):
     prompt=text
-    image = pipeline(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale,height=128,width=128).images[0]
+    with torch.no_grad():
+        image = pipeline(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale,height=128,width=128).images[0]
     return image
     
 mukh_biboron_app = gr.Interface(