Update app.py

app.py

@@ -13,13 +13,13 @@ model = MBartForConditionalGeneration.from_pretrained(model_name)
 def answer_question(text, question):
     # Prepare input text
     input_text = f"context: {text} question: {question}"
-    inputs = tokenizer(input_text, return_tensors="pt", max_length=1280000, truncation=False, padding="max_length")
+    inputs = tokenizer(input_text, return_tensors="pt", max_length=1024, truncation=False, padding="max_length")
     
     # Generate answer
     outputs = model.generate(
         input_ids=inputs["input_ids"],
         attention_mask=inputs["attention_mask"],
-        max_length=1280000,
+        max_length=1024,
         num_beams=5,
         early_stopping=True
     )