Update app.py

app.py

@@ -39,8 +39,7 @@ print("Loading model...")
 pipe = pipeline(
     "text-generation",
     # model="Qwen/Qwen2.5-0.5B-Instruct",
-    # model="Qwen/Qwen2.5-1.5B-Instruct",
-    model = "google/gemma-2-2b-it",
+    model="Qwen/Qwen2.5-1.5B-Instruct",
     device="cpu",
     torch_dtype="auto"
 )
@@ -219,15 +218,11 @@ def generate_response(user_prompt):
         4. TONE: Be concise, highly professional, and helpful.
         """
         
-        # Format for Qwen/Qwen2.5-1.5B-Instruct
-        # messages = [
-        #     {"role": "system", "content": system_prompt},
-        #     {"role": "user", "content": user_prompt}
-        # ]
-
-        # Format for google/gemma-2-2b-it
-        combined_prompt = f"{system_prompt}\n\nUser Question:\n{user_prompt}"
-        messages = [{"role": "user", "content": combined_prompt}]
+        messages = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt}
+        ]
+
         
         prompt = pipe.tokenizer.apply_chat_template(
             messages,