Update app.py

app.py

@@ -1,13 +1,14 @@
 import gradio as gr
 import torch
 import pandas as pd
-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)
 import bitsandbytes as bnb
 import evaluate
 import numpy as np
 import random
+import huggingface_hub
+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)
 
 def process(model_id, dataset):
     # Step 1: Load dataset
@@ -45,6 +46,8 @@ def process(model_id, dataset):
 
     #Step 5: Fine-tune the model
 
+    huggingface_hub.login()
+    
     model_id = model_id
     model = AutoModelForSequenceClassification.from_pretrained(model_id)