Gizachew commited on
Commit
ac2f881
·
verified ·
1 Parent(s): 675782e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -119,17 +119,17 @@ class AmharicPOSTagger:
119
  # Try to load from Hugging Face Hub
120
  try:
121
  model_path = hf_hub_download(
122
- repo_id=os.environ.get("HF_USERNAME", "your-username") + "/amharic-pos-tagger",
123
  filename="model.pth"
124
  )
125
  vocab_path = hf_hub_download(
126
- repo_id=os.environ.get("HF_USERNAME", "your-username") + "/amharic-pos-tagger",
127
  filename="vocab.pth"
128
  )
129
  except:
130
  # Fall back to local files
131
- model_path = "model.pth"
132
- vocab_path = "vocab.pth"
133
 
134
  # Load vocabularies
135
  vocabs = torch.load(vocab_path, map_location=self.device)
 
119
  # Try to load from Hugging Face Hub
120
  try:
121
  model_path = hf_hub_download(
122
+ repo_id=os.environ.get("HF_USERNAME", "Gizachew") + "/anduPOS",
123
  filename="model.pth"
124
  )
125
  vocab_path = hf_hub_download(
126
+ repo_id=os.environ.get("HF_USERNAME", "Gizachew") + "/anduPOS",
127
  filename="vocab.pth"
128
  )
129
  except:
130
  # Fall back to local files
131
+ model_path = "model_GRU_fold_1.pth"
132
+ vocab_path = "vocab_GRU_fold_1.pth"
133
 
134
  # Load vocabularies
135
  vocabs = torch.load(vocab_path, map_location=self.device)