Update app.py

app.py

@@ -23,7 +23,8 @@ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float
 #지운 옵션: device_map="auto"
 
 # KMMLU 데이터셋 로드
-dataset = load_dataset("HAERAE-HUB/KMMLU", "Accounting")
+#dataset = load_dataset("HAERAE-HUB/KMMLU", "Accounting")
+dataset = load_dataset("HAERAE-HUB/KMMLU")
 df = dataset['test'].to_pandas()
 
 def evaluate_model(question, choices):
@@ -62,10 +63,12 @@ def run_kmmlu_test(subject):
     summary = f"주제: {subject}\n정확도: {accuracy:.2%} ({correct}/{total})\n\n"
     return summary + "\n".join(results)
 
+subjects=df['subject'].unique().tolist()
+
 iface = gr.Interface(
     fn=run_kmmlu_test,
     inputs="Accounting",
-    #inputs=gr.Dropdown(choices=subjects, label="주제 선택"),
+    inputs=gr.Dropdown(choices=subjects, label="주제 선택"),
     outputs="text",
     title="Llama 3를 이용한 KMMLU 테스트",
     description="선택한 주제에 대해 KMMLU 테스트를 실행합니다."