Update app.py

app.py

@@ -139,7 +139,10 @@ def inference(model_name, prompt, guidance, steps, n_images=1, width=512, height
   if seed == 0:
     seed = random.randint(0, 2147483647)
 
+  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: