Update app.py

app.py

@@ -4,22 +4,20 @@ from diffusers import StableDiffusionPipeline
 
 
 #gr.Interface.load("models/bariscal/cbst_style")
-pipe = StableDiffusionPipeline.from_pretrained("bariscal/cbst_style", safety_checker=None) #, torch_dtype=torch.float16
+#pipe = StableDiffusionPipeline.from_pretrained("bariscal/cbst_style", safety_checker=None) #, torch_dtype=torch.float16
 
 
+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):
-    # Remove the torch.autocast("cuda") context manager
     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,
-                # Use the CPU for inference
-                generator="cpu"
-            ).images
-
-
+            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():