Update app.py

app.py

@@ -13,14 +13,14 @@ generator = torch.Generator(device='cpu')
 pipe = StableDiffusionPipeline.from_pretrained("bariscal/cbst_style", safety_checker=None)
 
 def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5):
-    with torch.inference_mode():
-        return pipe(
-            prompt, height=int(height), width=int(width),
-            negative_prompt=negative_prompt,
-            num_images_per_prompt=int(num_samples),
-            num_inference_steps=int(num_inference_steps), guidance_scale=guidance_scale,
-            generator="cpu"  # Use the CPU for inference
-        ).images
+    #with torch.inference_mode():
+    return pipe(
+        prompt, height=int(height), width=int(width),
+        negative_prompt=negative_prompt,
+        num_images_per_prompt=int(num_samples),
+        num_inference_steps=int(num_inference_steps), guidance_scale=guidance_scale,
+        generator="cpu"  # Use the CPU for inference
+    ).images
 
 with gr.Blocks() as demo:
     with gr.Row():