Update app.py

app.py

@@ -138,14 +138,24 @@ def app_outputs_by_model(uploaded_pdf, model_id, model, tokenizer, max_length, i
 
     else:  
         img_files, images, csv_files = [""]*max_imgboxes, [""]*max_imgboxes, [""]*max_imgboxes
-        img_files[0], img_files[1] = image_blank, image_blank
-        images[0], images[1] = Image.open(image_blank), Image.open(image_blank)
-        csv_file = "csv_wo_content.csv"
-        csv_files[0], csv_files[1] = gr.File.update(value=csv_file, visible=True), gr.File.update(value=csv_file, visible=True)
-        df, df_empty = dict(), pd.DataFrame()
-        df[0], df[1] = df_empty.to_csv(csv_file, encoding="utf-8", index=False), df_empty.to_csv(csv_file, encoding="utf-8", index=False)
+        if max_imgboxes >= 2:
+            img_files[0], img_files[1] = image_blank, image_blank
+            images[0], images[1] = Image.open(image_blank), Image.open(image_blank)
+            csv_file = "csv_wo_content.csv"
+            csv_files[0], csv_files[1] = gr.File.update(value=csv_file, visible=True), gr.File.update(value=csv_file, visible=True)
+            df, df_empty = dict(), pd.DataFrame()
+            df[0], df[1] = df_empty.to_csv(csv_file, encoding="utf-8", index=False), df_empty.to_csv(csv_file, encoding="utf-8", index=False)
+
+            return msg, img_files[0], img_files[1], images[0], images[1], csv_files[0], csv_files[1], df[0], df[1]
+        else:
+            img_files[0] = image_blank
+            images[0] = Image.open(image_blank)
+            csv_file = "csv_wo_content.csv"
+            csv_files[0] = gr.File.update(value=csv_file, visible=True)
+            df, df_empty = dict(), pd.DataFrame()
+            df[0] = df_empty.to_csv(csv_file, encoding="utf-8", index=False)          
     
-    return msg, img_files[0], images[0], csv_files[0], df[0]
+            return msg, img_files[0], images[0], csv_files[0], df[0]
 
 def app_outputs(uploaded_pdf):
     msg_lilt, img_files_lilt, images_lilt, csv_files_lilt, df_lilt = app_outputs_by_model(uploaded_pdf,