Update app.py

app.py

@@ -7,11 +7,17 @@ def generate_text(input_text, history):
     # Initialize the pipeline with specified parameters
     pipe = pipeline("text-generation", model="ngxson/MiniThinky-v2-1B-Llama-3.2", max_new_tokens=1024, temperature=0.7, do_sample=True)
     
-    # Prepare the messages
-    messages = [{"role": "user", "content": input_text}]
+    # Prepare the system message
+    system_message = "You are MiniThinky, a helpful AI assistant. You always think before giving the answer. Use <|thinking|> before thinking and <|answer|> before giving the answer."
+    
+    # Format the input messages
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": input_text}
+    ]
     
     # Generate response
-    response = pipe(messages)
+    response = pipe(messages, max_new_tokens=1024, temperature=0.7, do_sample=True)
     
     # Extract the generated text
     response_text = response[0]["generated_text"]