Update app.py

app.py

@@ -10,7 +10,7 @@ import os
 from datasets import Dataset, DatasetDict, load_dataset
 from peft import LoraConfig, PeftModel, prepare_model_for_kbit_training, get_peft_model
 from transformers import (AutoTokenizer, BitsAndBytesConfig, TrainingArguments, AutoModelForSequenceClassification, Trainer, EarlyStoppingCallback, DataCollatorWithPadding)
-from huggingface_hub import HfApi
+from huggingface_hub import login
 
 def process(model_id, dataset):
     # Step 1: Load dataset
@@ -48,8 +48,7 @@ def process(model_id, dataset):
 
     #Step 5: Fine-tune the model
 
-    api = HfApi()
-    api.login(token=os.environ["HF_TOKEN"])
+    login(token=os.environ.get("HF_TOKEN"))
     
     model_id = model_id
     model = AutoModelForSequenceClassification.from_pretrained(model_id)