Update app.py

app.py

@@ -6,6 +6,7 @@ import evaluate
 import numpy as np
 import random
 import huggingface_hub
+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)
@@ -46,7 +47,7 @@ def process(model_id, dataset):
 
     #Step 5: Fine-tune the model
 
-    huggingface_hub.login()
+    huggingface_hub.login(os.environ["OPENAI_API_KEY"])
     
     model_id = model_id
     model = AutoModelForSequenceClassification.from_pretrained(model_id)