Update app.py

app.py

@@ -27,8 +27,7 @@ def replace_single_quotes(text):
     return replaced_text
 
 class PromptEnhancerV2:
-    def __init__(self, models_root_path, device_map="cuda", torch_dtype="bfloat16"):#auto
-        device_map = "cuda:0"
+    def __init__(self, models_root_path, device_map="auto", torch_dtype="bfloat16"):#auto
         if not logging.getLogger(__name__).handlers:
             logging.basicConfig(level=logging.INFO)
         self.logger = logging.getLogger(__name__)