Update app.py

app.py

@@ -23,9 +23,9 @@ output: Potential widening of the achievement gap if data is not used equitably.
 def generate(text):
     try:
         prompt = system_prompt + f"\ninput: {text}\noutput:"
-        inputs = tokenizer(prompt, return_tensors="pt", return_attention_mask=True)
-        input_ids = inputs["input_ids"].unsqueeze(0)
-        attention_mask = inputs["attention_mask"].unsqueeze(0)
+        encoding = tokenizer(prompt, return_tensors="pt", return_attention_mask=True, truncation=True)
+        input_ids = encoding["input_ids"].unsqueeze(0)
+        attention_mask = encoding["attention_mask"].unsqueeze(0)
         outputs = model.generate(input_ids, attention_mask=attention_mask, max_length=256)
         response_text = tokenizer.decode(outputs[0], skip_special_tokens=True).split("output:")[-1].strip()
         return response_text if response_text else "No valid response generated."