Update app.py

app.py

@@ -155,24 +155,24 @@ def real_time_check(image_file):
     try:
         ko_deplot_generated_df, ko_deplot_generated_title=ko_deplot_convert_to_dataframe(ko_deplot_generated_txt)
         gpt_generated_df, gpt_generated_title=gpt_convert_to_dataframe(gpt_generated_txt)
-        return gr.DataFrame(ko_deplot_generated_df, label= ko_deplot_generated_title), gr.DataFrame(gpt_generated_df, label= gpt_generated_title), None,None,0
+        return ko_deplot_generated_df, ko_deplot_generated_title, gpt_generated_df, gpt_generated_title, None,None,0
     except Exception as e:
-        return None,None,ko_deplot_generated_txt,gpt_generated_txt,1
+        return None,None,None,None,ko_deplot_generated_txt,gpt_generated_txt,1
     
 flag = 0 #flag to check whether exception happens or not. if flag is 1, it means that exception(generated txt cannot be converted to pandas dataframe) happens. 
 def inference(image_uploader,mode_selector):
     if(mode_selector=="파일 업로드"):
-        ko_deplot_generated_df, gpt_generated_df,ko_deplot_generated_txt, gpt_generated_txt, flag= real_time_check(image_uploader)
+        ko_deplot_generated_df, ko_deplot_generated_title, gpt_generated_df, gpt_generated_title, ko_deplot_generated_txt, gpt_generated_txt, flag= real_time_check(image_uploader)
         if flag==1:
             return gr.update(visible=False), gr.update(visible=False), gr.Text(ko_deplot_generated_txt,visible=True),gr.Text(gpt_generated_txt,visible=True)
         else:
-            return ko_deplot_generated_df, gpt_generated_df, gr.update(visible=False),gr.update(visible=False)
+            return gr.DataFrame(ko_deplot_generated_df, label = ko_deplot_generated_title, visible=True), gr.DataFrame(gpt_generated_df, label = gpt_generated_title, visible=True), gr.update(visible=False),gr.update(visible=False)
     else:
-        ko_deplot_generated_df, gpt_generated_df,ko_deplot_generated_txt, gpt_generated_txt, flag= real_time_check(image_files[current_image_index])
+        ko_deplot_generated_df, ko_deplot_generated_title, gpt_generated_df, gpt_generated_title, ko_deplot_generated_txt, gpt_generated_txt, flag= real_time_check(image_files[current_image_index])
         if flag==1:
             return gr.update(visible=False), gr.update(visible=False), gr.Text(ko_deplot_generated_txt,visible=True),gr.Text(gpt_generated_txt,visible=True)
         else:
-            return gr.DataFrame(ko_deplot_generated_df,visible=True), gr.DataFrame(gpt_generated_df,visible=True), gr.update(visible=False),gr.update(visible=False)
+            return gr.DataFrame(ko_deplot_generated_df, label = ko_deplot_generated_title, visible=True), gr.DataFrame(gpt_generated_df, label = gpt_generated_title, visible=True), gr.update(visible=False),gr.update(visible=False)
         
 def toggle_model(selected_models,flag):
     # Create a visibility list initialized to False for all components