Update app.py

app.py

@@ -5,7 +5,7 @@ from openai import OpenAI
 import os
 import torch
 from peft import LoraConfig, PeftModel, prepare_model_for_kbit_training, get_peft_model
-from dotenv import load_dotenv
+from huggingface_hub import login
 
 # Load environment variables
 load_dotenv()
@@ -16,8 +16,12 @@ client = OpenAI(
     api_key=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
 )
 
-from huggingface_hub import login
-login(token="HUGGINGFACEHUB_API_TOKEN")
+api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+
+if api_token:
+    login(token=api_token)
+else:
+    print("API token is not set in the environment variables.")
 
 # Define model links and configurations
 model_links = {