Update app.py

app.py

@@ -966,7 +966,7 @@ def update_results(selected_models):
     visibility = [False] * 9
 
     # Update visibility based on the selected models
-    if "ko_deplot" in selected_models:
+    if "VAIV_DePlot" in selected_models:
         visibility[0] = True  # ko_deplot_generated_table
         visibility[1] = True  # ko_deplot_score_table
         visibility[6] = True  # ko_deplot_label_table
@@ -976,7 +976,7 @@ def update_results(selected_models):
         visibility[3] = True  # aihub_deplot_score_table
         visibility[7] = True  # aihub_deplot_label_table
     '''
-    if "unichart" in selected_models:
+    if "VAIV_UniChart" in selected_models:
         visibility[4] = True  # unichart_generated_table
         visibility[5] = True  # unichart_score_table
         visibility[8] = True  # unichart_label_table
@@ -1067,7 +1067,7 @@ with gr.Blocks(css=css) as iface:
             file_uploader= gr.File(file_count="single", file_types=[".txt"], visible=False)
             file_upload_option=gr.Radio(choices=["low score 차트","high score 차트"],label="파일 업로드 옵션",visible=False)
             chart_type = gr.Dropdown(["일반 가로 막대형","누적 가로 막대형","100% 기준 누적 가로 막대형", "일반 세로 막대형","누적 세로 막대형","100% 기준 누적 세로 막대형","선형", "원형", "기타 방사형", "기타 혼합형", "전체"], label="Chart Type", value="all")
-            model_type=gr.Dropdown(["vaiv_deplot","unichart","all"],label="model",multiselect=True)
+            model_type=gr.Dropdown(["VAIV_DePlot","VAIV_UniChart","all"],label="model",multiselect=True)
             image_displayer=gr.Image(visible=True)
             with gr.Row():
                 pre_button=gr.Button("이전",interactive="False")
@@ -1077,17 +1077,17 @@ with gr.Blocks(css=css) as iface:
             #file_button.click(interface_selector, inputs=gr.State("파일 업로드"), outputs=[image_uploader, file_uploader,mode,mode_label,image_name])
             inference_button=gr.Button("추론")
         with gr.Column():
-            ko_deplot_generated_table=gr.DataFrame(visible=False,label="vaiv-deplot 추론 결과",elem_classes="dataframe-class")
+            ko_deplot_generated_table=gr.DataFrame(visible=False,label="VAIV_DePlot 추론 결과",elem_classes="dataframe-class")
             aihub_deplot_generated_table=gr.DataFrame(visible=False,label="aihub-deplot 추론 결과",elem_classes="dataframe-class")
-            unichart_generated_table=gr.DataFrame(visible=False,label="unichart 추론 결과",elem_classes="dataframe-class")
+            unichart_generated_table=gr.DataFrame(visible=False,label="VAIV_UniChart 추론 결과",elem_classes="dataframe-class")
         with gr.Column():
-            ko_deplot_label_table=gr.DataFrame(visible=False,label="vaiv-deplot 정답테이블",elem_classes="dataframe-class")
+            ko_deplot_label_table=gr.DataFrame(visible=False,label="VAIV_DePlot 정답테이블",elem_classes="dataframe-class")
             aihub_deplot_label_table=gr.DataFrame(visible=False,label="aihub-deplot 정답테이블",elem_classes="dataframe-class")
-            unichart_label_table=gr.DataFrame(visible=False,label="unichart 정답테이블",elem_classes="dataframe-class")
+            unichart_label_table=gr.DataFrame(visible=False,label="VAIV_UniChart 정답테이블",elem_classes="dataframe-class")
         with gr.Column():
-            ko_deplot_score_table=gr.DataFrame(visible=False,label="vaiv_deplot 점수",elem_classes="dataframe-class")
+            ko_deplot_score_table=gr.DataFrame(visible=False,label="VAIV_DePlot 점수",elem_classes="dataframe-class")
             aihub_deplot_score_table=gr.DataFrame(visible=False,label="aihub_deplot 점수",elem_classes="dataframe-class")
-            unichart_score_table=gr.DataFrame(visible=False,label="unichart 점수",elem_classes="dataframe-class")
+            unichart_score_table=gr.DataFrame(visible=False,label="VAIV_UniChart 점수",elem_classes="dataframe-class")
     model_type.change(
                         update_results,
                         inputs=[model_type],