Upload 2 files

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import os
+from huggingface_hub import login
 from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer
 from transformers import AutoTokenizer
 
@@ -17,6 +18,7 @@ model_name = "roberta-base"
 learning_rate = 5e-5
 batch_size_train = 64
 step = 1900
+login(token='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')
 
 file_name = "model-{}.pt".format(step)
 state_dict = torch.load(os.path.join(file_name))

requirements.txt

@@ -1,3 +1,4 @@
 gradio
 torch
-transformers
+transformers
+huggingface_hub