Update app.py

app.py

@@ -66,10 +66,9 @@ class MentalHealthChatbot:
         # Tokenize the prompt
         inputs = self.tokenizer(full_prompt, return_tensors="pt", padding=True, truncation=True)
         
-        # Generate response
         outputs = self.model.generate(
             **inputs,
-            max_length=max_tokens,
+            max_length=max(512, inputs["input_ids"].shape[1] + max_tokens)
             temperature=temperature,
             top_p=top_p,
             no_repeat_ngram_size=3,