aagamjtdev commited on
Commit
77ead37
·
1 Parent(s): c08f202
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -6,10 +6,9 @@ import sys
6
  from datetime import datetime
7
  import shutil
8
 
9
- # FIX: Update the script name to the correct one you uploaded
10
  TRAINING_SCRIPT = "HF_LayoutLM_with_Passage.py"
11
 
12
- # --- CORRECTED MODEL PATH BASED ON YOUR SCRIPT ---
13
  MODEL_OUTPUT_DIR = "checkpoints"
14
  MODEL_FILE_NAME = "layoutlmv3_crf_passage.pth"
15
  MODEL_FILE_PATH = os.path.join(MODEL_OUTPUT_DIR, MODEL_FILE_NAME)
@@ -296,7 +295,6 @@ with gr.Blocks(title="LayoutLMv3 Fine-Tuning App", theme=gr.themes.Soft()) as de
296
  outputs=[model_download]
297
  )
298
 
299
- # Add example info
300
  gr.Markdown(
301
  """
302
  ---
 
6
  from datetime import datetime
7
  import shutil
8
 
9
+
10
  TRAINING_SCRIPT = "HF_LayoutLM_with_Passage.py"
11
 
 
12
  MODEL_OUTPUT_DIR = "checkpoints"
13
  MODEL_FILE_NAME = "layoutlmv3_crf_passage.pth"
14
  MODEL_FILE_PATH = os.path.join(MODEL_OUTPUT_DIR, MODEL_FILE_NAME)
 
295
  outputs=[model_download]
296
  )
297
 
 
298
  gr.Markdown(
299
  """
300
  ---