Update app.py

app.py

@@ -1,8 +1,18 @@
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelForMultipleChoice, AutoModelForQuestionAnswering
+from torch.utils.data import DataLoader # <--- 新增這一行
+from transformers import (
+    AutoTokenizer, 
+    AutoModelForMultipleChoice, 
+    AutoModelForQuestionAnswering,
+    default_data_collator # 如果您在 app.py 中也使用它
+)
 import json
-import collections # 如果您的 postprocess_qa_predictions 需要
+import collections 
+import numpy as np
+from datasets import Dataset 
+
+from utils_qa import postprocess_qa_predictions 
 
 # 假設 utils_qa.py 在同一目錄下 (或者您需要將其函數複製過來或確保可導入)
 # from utils_qa import postprocess_qa_predictions # 您可能需要完整路徑或將其放入 requirements.txt