Clean app.py startup without spaces conflict

app.py

@@ -17,16 +17,12 @@ from minirewrite.pipeline import MiniRewrite
 
 # Global model instance
 print("Initializing MiniRewrite model...")
-# Determine weights location: local folder or runs/rtx4060 fallback
-if (ROOT_DIR / "tokenizer.json").exists() and (ROOT_DIR / "best.pt").exists():
-    MODEL_DIR = ROOT_DIR
-elif (ROOT_DIR / "runs" / "rtx4060" / "tokenizer.json").exists():
-    MODEL_DIR = ROOT_DIR / "runs" / "rtx4060"
-else:
-    MODEL_DIR = ROOT_DIR
-
-pipeline = MiniRewrite.from_pretrained(MODEL_DIR, checkpoint_name="best.pt", device="auto")
-print(f"Model loaded on {pipeline.device}!")
+MODEL_DIR = ROOT_DIR if (ROOT_DIR / "best.pt").exists() else (ROOT_DIR / "runs" / "rtx4060")
+
+# CPU inference is extremely fast (~50ms) for this 7.6M model and requires zero GPU cold-start
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline = MiniRewrite.from_pretrained(MODEL_DIR, checkpoint_name="best.pt", device=device)
+print(f"Model loaded successfully on {pipeline.device}!")
 
 
 def process_rewrite(text: str, task: str, tone: str, style: str):