Update app.py

app.py

@@ -20,7 +20,7 @@ base_model_path = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
 base_cache = "model-cache"
 vae_model_path = "stabilityai/sd-vae-ft-mse"
 ip_cache = "./ip-cache"
-device = "cuda"
+#device = "cuda"
 
 # Setup function to load models and other dependencies
 def setup():
@@ -57,13 +57,12 @@ def setup():
         safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker"),
         cache_dir=base_cache,
     )
-    pipe = pipe.to(device)
+    #pipe = pipe.to(device)
     
     # IP adapter
     ip_model = IPAdapterFaceID(
         pipe,
-        "ip-cache/ip-adapter-faceid_sd15.bin",
-        device
+        "ip-cache/ip-adapter-faceid_sd15.bin" #device
     )
     
     return app, ip_model
@@ -134,7 +133,7 @@ def construct_prompt(base_prompt, additional_prompt, gender):
     
     return full_prompt
 
-@spaces.GPU(duration = 40)
+#@spaces.GPU(duration = 40)
 def generate_image(face_image_1, face_image_2, additional_prompt, gender):
     base_prompt = "portrait of a 6 y.o. child, 8k, HD, happy, perfect eyes, cute"
     full_prompt = construct_prompt(base_prompt, additional_prompt, gender)