Update app.py

app.py

@@ -9,14 +9,17 @@ hf_token = os.getenv("HF_TOKEN")
 MODEL_ID = "Qwen/Qwen2.5-Coder-1.0B-Instruct"
 
 print("Loading pipeline (model + tokenizer)...")
+
+generator = pipeline("text-generation", model="Qwen/Qwen2.5-Coder-1.5B-Instruct")
+
 # The pipeline automatically downloads everything needed
-generator = pipeline(
-    "text-generation",
-    model=MODEL_ID,
-    device_map="cpu", # Force CPU for free-tier Spaces
-    torch_dtype="auto",
-    token=hf_token
-)
+# generator = pipeline(
+#     "text-generation",
+#     model=MODEL_ID,
+#     device_map="cpu", # Force CPU for free-tier Spaces
+#     torch_dtype="auto",
+#     token=hf_token
+# )
 print("Pipeline loaded successfully")
 
 app = Flask(__name__)