thak123 commited on
Commit
db99934
1 Parent(s): 092ad3f

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +4 -4
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
- print(hf_hub_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier",filename = "pytorch_model.bin"))
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']))