Update app.py

app.py

@@ -4,8 +4,8 @@
 # in the Software without restriction, including without limitation the rights
 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 # copies of the Software, and to permit persons to whom the Software is
-import spaces
 
+import spaces
 import os
 import random
 import uuid
@@ -17,7 +17,6 @@ from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 from typing import Tuple
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-
 css = '''
 .gradio-container{max-width: 570px !important}
 h1{text-align:center}
@@ -45,7 +44,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
 
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+device = "cpu")
 
 style_list = [
     {
@@ -100,13 +99,7 @@ def load_and_prepare_model(model_id):
         add_watermarker=False,
     ).to(device)
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    
-    if USE_TORCH_COMPILE:
-        pipe.compile()
-    
-    if ENABLE_CPU_OFFLOAD:
-        pipe.enable_model_cpu_offload()
-    
+
     return pipe
 
 # Preload and compile both models