Update app.py

app.py

@@ -111,7 +111,10 @@ def error_str(error, title="Error"):
             {error}"""  if error else ""
 
 def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
-  generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
+  if torch.cuda.is_available():
+    generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
+  else:
+    generator = torch.Generator.manual_seed(seed) if seed != 0 else None
   prompt = f"{prefix} {prompt}" if auto_prefix else prompt
     
   try: