Update app.py

app.py

@@ -10,6 +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
 
 def process(model_id, dataset):
     # Step 1: Load dataset
@@ -47,7 +48,8 @@ def process(model_id, dataset):
 
     #Step 5: Fine-tune the model
 
-    huggingface_hub.login(os.environ["OPENAI_API_KEY"])
+    api = HfApi()
+    api.login(token=os.environ["HF_TOKEN"])
     
     model_id = model_id
     model = AutoModelForSequenceClassification.from_pretrained(model_id)