fix

pages/batch_evaluation.py

@@ -52,7 +52,7 @@ def process(seed, model_selection, prompt, num=10):
     for data in random.choices(dataset, k=num):
         dialogue = data['dialogue']
         summary = data['summary']
-        response = generate_answer(dialogue, model, model_selection, prompt)
+        response = generate_answer(dialogue, model_selection, prompt)
 
         rouge_score = metric_rouge_score(response, summary)
 

pages/summarization_playground.py

@@ -31,7 +31,7 @@ Back in Boston, Kidd is going to rely on Lively even more. He'll play close to 3
     random_label: ""
 }
 
-def generate_answer(sources, model, model_name, prompt):
+def generate_answer(sources, model_name, prompt):
     content = prompt + '\n' + sources + '\n\n'
     global __model_on_gpu__
 
@@ -50,7 +50,7 @@ def generate_answer(sources, model, model_name, prompt):
 
 def process_input(input_text, model_selection, prompt):
     if input_text:
-        response = generate_answer(input_text, model, model_selection, prompt)
+        response = generate_answer(input_text, model_selection, prompt)
         return f"## Original Article:\n\n{input_text}\n\n## Summarization:\n\n{response}"
     else:
         return "Please fill the input to generate outputs."