Update app.py
Browse files
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", "
|
| 123 |
filename="model.pth"
|
| 124 |
)
|
| 125 |
vocab_path = hf_hub_download(
|
| 126 |
-
repo_id=os.environ.get("HF_USERNAME", "
|
| 127 |
filename="vocab.pth"
|
| 128 |
)
|
| 129 |
except:
|
| 130 |
# Fall back to local files
|
| 131 |
-
model_path = "
|
| 132 |
-
vocab_path = "
|
| 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)
|