Gleb Vinarskis
commited on
Commit
·
6cd3d12
1
Parent(s):
708a64a
new model
Browse files- impresso_langident_wrapper.py +0 -10
- pytorch_model.bin +2 -2
impresso_langident_wrapper.py
CHANGED
@@ -6,16 +6,6 @@ from huggingface_hub import hf_hub_download
|
|
6 |
|
7 |
|
8 |
class Pipeline_One(Pipeline):
|
9 |
-
def __init__(self, model_name="pytorch_model.bin", repo_id="Maslionok/pipeline1", revision="main", **kwargs):
|
10 |
-
"""
|
11 |
-
Initialize the pipeline and manually load the Floret model from Hugging Face.
|
12 |
-
"""
|
13 |
-
super().__init__(**kwargs) # ✅ Call parent class constructor
|
14 |
-
|
15 |
-
# ✅ Manually download and load the Floret model
|
16 |
-
model_path = hf_hub_download(repo_id=repo_id, filename=model_name, revision=revision)
|
17 |
-
self.model = floret.load_model(model_path)
|
18 |
-
|
19 |
# def __init__(self, model_path: str):
|
20 |
# """
|
21 |
# Initialize the Floret language detection pipeline
|
|
|
6 |
|
7 |
|
8 |
class Pipeline_One(Pipeline):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
# def __init__(self, model_path: str):
|
10 |
# """
|
11 |
# Initialize the Floret language detection pipeline
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3496a508a9a3511c8a55e4d0e6f471c70c68c2a8c4784b3b2b5dc16ffb87d238
|
3 |
+
size 714314041
|