Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
+import torch
 
 
 gr.Interface.load("models/bariscal/cbst_style")
 
 
 def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5):
-    with torch.autocast("cuda"), torch.inference_mode():
+    with torch.Generator('cpu'), torch.inference_mode():
         return pipe(
                 prompt, height=int(height), width=int(width),
                 negative_prompt=negative_prompt,