FlameF0X commited on
Commit
a8a99c3
·
verified ·
1 Parent(s): 9e0dda3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -202,7 +202,7 @@ class HybridBertModel(nn.Module):
202
  # 2. INITIALIZATION
203
  # ============================================================================
204
 
205
- REPO_ID = "FlameF0X/i3-BERT-v2"
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 'FlameF0X/i3-BERT-v2'")
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