Update app.py

app.py

@@ -8,15 +8,14 @@ pipeline = DiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
      text_encoder = AutoModel.from_pretrained("csebuetnlp/banglabert"),
      custom_pipeline="gr33nr1ng3r/Mukh-Oboyob",
-     no_safe_serialization=True
+     
 )
 pipeline.unet.load_attn_procs("gr33nr1ng3r/Mukh-Oboyob")
 
 
 def diffusion(text,num_inference_steps,guidance_scale):
     prompt=text
-    with torch.no_grad():
-        image = pipeline(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale,height=128,width=128).images[0]
+    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(