Spaces:
Running
Running
Commit
·
77ead37
1
Parent(s):
c08f202
change
Browse files
app.py
CHANGED
|
@@ -6,10 +6,9 @@ import sys
|
|
| 6 |
from datetime import datetime
|
| 7 |
import shutil
|
| 8 |
|
| 9 |
-
|
| 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 |
---
|