Update app.py
Browse files
app.py
CHANGED
|
@@ -202,7 +202,7 @@ class HybridBertModel(nn.Module):
|
|
| 202 |
# 2. INITIALIZATION
|
| 203 |
# ============================================================================
|
| 204 |
|
| 205 |
-
REPO_ID = "
|
| 206 |
MODEL_FILENAME = "i3-bert.pt"
|
| 207 |
TOKENIZER_FILENAME = "tokenizer_bert.json"
|
| 208 |
|
|
@@ -212,7 +212,7 @@ try:
|
|
| 212 |
tokenizer_path = hf_hub_download(repo_id=REPO_ID, filename=TOKENIZER_FILENAME)
|
| 213 |
except Exception as e:
|
| 214 |
print(f"Error downloading files: {e}")
|
| 215 |
-
print("Ensure 'i3-bert.pt' and 'tokenizer_bert.json' exist in '
|
| 216 |
raise e
|
| 217 |
|
| 218 |
# Load Tokenizer
|
|
|
|
| 202 |
# 2. INITIALIZATION
|
| 203 |
# ============================================================================
|
| 204 |
|
| 205 |
+
REPO_ID = "i3-lab/i3-BERT-v2"
|
| 206 |
MODEL_FILENAME = "i3-bert.pt"
|
| 207 |
TOKENIZER_FILENAME = "tokenizer_bert.json"
|
| 208 |
|
|
|
|
| 212 |
tokenizer_path = hf_hub_download(repo_id=REPO_ID, filename=TOKENIZER_FILENAME)
|
| 213 |
except Exception as e:
|
| 214 |
print(f"Error downloading files: {e}")
|
| 215 |
+
print("Ensure 'i3-bert.pt' and 'tokenizer_bert.json' exist in 'i3-lab/i3-BERT-v2'")
|
| 216 |
raise e
|
| 217 |
|
| 218 |
# Load Tokenizer
|