Update app.py

app.py

@@ -25,7 +25,7 @@ max_length = 512
 max_target_length = 200
 
 # Load your dataset
-dataset = datasets.load_dataset("minh21/COVID-QA-Chunk-64-testset-biencoder-data-90_10", split="test")
+dataset = datasets.load_dataset("minh21/COVID-QA-Chunk-64-testset-biencoder-data-90_10", split="train")
 dataset = dataset.shuffle()
 dataset = dataset.select(range(5))
 
@@ -71,6 +71,8 @@ def retrieve_context(question, contexts):
 
 # Define your function to generate answers
 def generate_answer(question, context, contexts):
+    if type(contexts) is str:
+        contexts = contexts.split(',')
     context = retrieve_context(question, contexts)
     
     # Combine question and context