Spaces:

ngupta2026
/

Gen_AI_Project

Sleeping

ngupta2026 commited on Apr 28

Commit

f8fd50e

verified ·

1 Parent(s): 454bba9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ id2label = {v:k for k,v in label2id.items()}
 # =========================
 # LOAD YOUR TRAINED MODEL
 # =========================
-MODEL_NAME = "ngupta2026/sroie-layoutlm-v2"   # 🔥 your new model
 model = LayoutLMForTokenClassification.from_pretrained(MODEL_NAME)
 tokenizer = LayoutLMTokenizerFast.from_pretrained(MODEL_NAME)

 # =========================
 # LOAD YOUR TRAINED MODEL
 # =========================
+MODEL_NAME = "ngupta2026/sroie-layoutlm"   # 🔥 your new model
 model = LayoutLMForTokenClassification.from_pretrained(MODEL_NAME)
 tokenizer = LayoutLMTokenizerFast.from_pretrained(MODEL_NAME)