Update app.py

app.py

@@ -123,7 +123,10 @@ def inference(model_name, prompt, guidance, steps, n_images=1, width=512, height
   if seed == 0:
     seed = random.randint(0, 2147483647)
 
-  generator = torch.Generator('cuda').manual_seed(seed)
+  if torch.cuda.is_available():
+    generator = torch.Generator('cuda').manual_seed(seed)
+  else:
+    generator = torch.Generator().manual_seed(seed)
 
   try:
     if img is not None: