Update app.py

app.py

@@ -167,14 +167,11 @@ def generate_60(
     num_images: int = 1,  
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
-    torch.set_default_device('cuda')
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cuda').manual_seed(seed)
-
+    generator = torch.Generator(device='cpu').manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
-
     options = {
         "prompt": [prompt] * num_images,
         "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
@@ -185,10 +182,8 @@ def generate_60(
         "generator": generator,
         "output_type": "pil",
     }
-
     if use_resolution_binning:
         options["use_resolution_binning"] = True
-
     images = []
     with torch.no_grad():
       for i in range(0, num_images, BATCH_SIZE):
@@ -205,7 +200,6 @@ def generate_60(
     gc.collect()
     return image_paths, seed
 
-
 @spaces.GPU(duration=90)
 def generate_90(
     model_choice: str,
@@ -226,10 +220,8 @@ def generate_90(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cuda').manual_seed(seed)
-
+    generator = torch.Generator(device='cpu').manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
-
     options = {
         "prompt": [prompt] * num_images,
         "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
@@ -240,10 +232,8 @@ def generate_90(
         "generator": generator,
         "output_type": "pil",
     }
-
     if use_resolution_binning:
         options["use_resolution_binning"] = True
-
     images = []
     with torch.no_grad():
       for i in range(0, num_images, BATCH_SIZE):
@@ -274,23 +264,7 @@ def load_predefined_images1():
     ]
     return predefined_images1
 
-
-# def load_predefined_images():
-#     predefined_images = [
-#         "assets2/11.png",
-#         "assets2/22.png",
-#         "assets2/33.png",
-#         "assets2/44.png",
-#         "assets2/55.png",
-#         "assets2/66.png",
-#         "assets2/77.png",
-#         "assets2/88.png",
-#         "assets2/99.png",
-#     ]
-#     return predefined_image
-
-
-with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
+with gr.Blocks(css=css) as demo:
     gr.Markdown(DESCRIPTIONXX)
     with gr.Row():
         prompt = gr.Text(