Update app.py

app.py

@@ -24,7 +24,7 @@ except Exception as e:
 
 # 현재 스크립트의 디렉토리를 기준으로 상대 경로 설정
 current_dir = os.path.dirname(os.path.abspath(__file__))
-parquet_path = os.path.join(current_dir, 'train-00000-of-00005.parquet')
+parquet_path = os.path.join(current_dir, 'train-00000-of-00001.parquet')
 
 # Parquet 파일 로드
 try:
@@ -34,7 +34,7 @@ try:
     print(f"컬럼: {df.columns}")
 except Exception as e:
     print(f"Parquet 파일 로드 중 오류 발생: {e}")
-    df = pd.DataFrame(columns=['question', 'answer'])  # 빈 DataFrame 생성
+    df = pd.DataFrame(columns=['instruction', 'response_a'])  # 빈 DataFrame 생성
 
 def get_answer(question):
     matching_answer = df[df['question'] == question]['answer'].values