Spaces:
Runtime error
Runtime error
File size: 636 Bytes
c8eed0a f71a77c 4b65f9f db99934 f71a77c c8eed0a db99934 93ab1cf 2de6916 93ab1cf db99934 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import os
from huggingface_hub import hf_hub_download,snapshot_download
# print(snapshot_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier"))
model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier",filename = "pytorch_model.bin")
print(os.listdir())
model_name = model_link #"FFZG-cleopatra/Croatian-News-Classifier"
output_path = ""
drop_out = 0.3
max_length = 512
epochs = 5
label2id = {'neutral': 0, 'negative': 1, 'positive': 2}
id2label = {0: 'neutral', 1: 'negative', 2: 'positive'}
output_dir=""
batch_size=16 #32
learning_rate=2e-5
# from pip import _internal
# print(_internal.main(['list']))
|