Spaces:
Runtime error
Runtime error
Update config.py
Browse files
config.py
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
import os
|
2 |
from huggingface_hub import hf_hub_download,snapshot_download
|
3 |
# print(snapshot_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier"))
|
4 |
-
|
5 |
|
6 |
print(os.listdir())
|
7 |
-
model_name = "FFZG-cleopatra/Croatian-News-Classifier"
|
8 |
|
9 |
output_path = ""
|
10 |
drop_out = 0.3
|
@@ -16,5 +16,5 @@ output_dir=""
|
|
16 |
batch_size=16 #32
|
17 |
learning_rate=2e-5
|
18 |
|
19 |
-
from pip import _internal
|
20 |
-
print(_internal.main(['list']))
|
|
|
1 |
import os
|
2 |
from huggingface_hub import hf_hub_download,snapshot_download
|
3 |
# print(snapshot_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier"))
|
4 |
+
model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier",filename = "pytorch_model.bin")
|
5 |
|
6 |
print(os.listdir())
|
7 |
+
model_name = model_link #"FFZG-cleopatra/Croatian-News-Classifier"
|
8 |
|
9 |
output_path = ""
|
10 |
drop_out = 0.3
|
|
|
16 |
batch_size=16 #32
|
17 |
learning_rate=2e-5
|
18 |
|
19 |
+
# from pip import _internal
|
20 |
+
# print(_internal.main(['list']))
|