mainss

.gitignore

@@ -66,4 +66,8 @@ venv.bak/
 
 # IDEs
 .idea/
-.vscode/
+.vscode/
+
+# Custom
+model/
+upload_model.py

app.py

@@ -5,23 +5,30 @@ import torch
 import numpy as np
 
 # --- CONFIGURATION ---
-MODEL_DIR = "./model"
+MODEL_REPO_ID = "Nomi78600/bert-ner-squad"
 st.set_page_config(page_title="NER with BERT", page_icon="🤖", layout="wide")
 
 # --- MODEL LOADING ---
 @st.cache_resource
-def load_model_and_tokenizer(model_path):
-    """Load the fine-tuned model and tokenizer from a local directory."""
+def load_model_and_tokenizer(repo_id):
+    """Load the fine-tuned model and tokenizer from the Hugging Face Hub."""
     try:
-        tokenizer = AutoTokenizer.from_pretrained(model_path)
-        model = AutoModelForTokenClassification.from_pretrained(model_path)
+        tokenizer = AutoTokenizer.from_pretrained(repo_id)
+        model = AutoModelForTokenClassification.from_pretrained(repo_id)
         return tokenizer, model
+    except OSError:
+        st.error(f"Error loading model: Could not find repository '{repo_id}' on the Hugging Face Hub. Please check the name.")
+        return None, None
     except Exception as e:
-        st.error(f"Error loading model: {e}")
+        st.error(f"An unexpected error occurred while loading the model: {e}")
         return None, None
 
-tokenizer, model = load_model_and_tokenizer(MODEL_DIR)
+# Load the model from the Hub
+with st.spinner(f"Loading model '{MODEL_REPO_ID}' from Hugging Face Hub..."):
+    tokenizer, model = load_model_and_tokenizer(MODEL_REPO_ID)
+
 if model is None:
+    st.error("Application failed to load the model. Please check the logs.")
     st.stop()
 
 # --- NER VISUALIZATION ---

model/config.json

@@ -9,28 +9,28 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4",
-    "5": "LABEL_5",
-    "6": "LABEL_6",
-    "7": "LABEL_7",
-    "8": "LABEL_8"
+    "0": "O",
+    "1": "B-PER",
+    "2": "I-PER",
+    "3": "B-ORG",
+    "4": "I-ORG",
+    "5": "B-LOC",
+    "6": "I-LOC",
+    "7": "B-MISC",
+    "8": "I-MISC"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4,
-    "LABEL_5": 5,
-    "LABEL_6": 6,
-    "LABEL_7": 7,
-    "LABEL_8": 8
+    "O": 0,
+    "B-PER": 1,
+    "I-PER": 2,
+    "B-ORG": 3,
+    "I-ORG": 4,
+    "B-LOC": 5,
+    "I-LOC": 6,
+    "B-MISC": 7,
+    "I-MISC": 8
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,

requirements.txt

@@ -4,4 +4,5 @@ streamlit
 datasets
 seqeval
 numpy
-scipy
+scipy
+huggingface-hub