Update app.py

app.py

@@ -145,6 +145,10 @@ def app_outputs_by_model(uploaded_pdf, model_id, model, tokenizer, max_length, i
             csv_files.append(gr.File.update(value=csv_file, visible=True))
             df[i].to_csv(csv_file, encoding="utf-8", index=False)
 
+        if max_imgboxes >= 2:
+            return msg, img_files[0], img_files[1], images[0], images[1], csv_files[0], csv_files[1], df[0], df[1]
+        else:    
+            return msg, img_files[0], images[0], csv_files[0], df[0]
     else:  
         img_files, images, csv_files = [""]*max_imgboxes, [""]*max_imgboxes, [""]*max_imgboxes
         if max_imgboxes >= 2: