darinanina dina no

app.py

@@ -1,5 +1,4 @@
 from transformers import MBartForConditionalGeneration, MBart50Tokenizer
-import dat
 import gradio as gr
 
 # Load the model and tokenizer
@@ -11,9 +10,9 @@ model = MBartForConditionalGeneration.from_pretrained(model_name)
 
 
 
-def answer_question(context, question):
+def answer_question(text, question):
     # Prepare input text
-    input_text = f"context: {context} question: {question}"
+    input_text = f"context: {text} question: {question}"
     inputs = tokenizer(input_text, return_tensors="pt", max_length=1280000, truncation=False, padding="max_length")
     
     # Generate answer
@@ -30,9 +29,9 @@ def answer_question(context, question):
     return answer
 
 demo = gr.Interface(
-    fn = answer_question,
-    inputs = ['context', 'question'],
-    outputs = ['text']
+  fn=answer_question,
+  inputs=["text", "question"],
+  outputs=["text"]
 )