Update app.py

app.py

@@ -2,6 +2,17 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
+from huggingface_hub import HfApi, HfFolder
+
+# Replace '<your_api_token>' with your actual API token
+api_token = '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'
+
+# Initialize the HfApi with the API token
+api = HfApi(token=api_token)
+print('loggedin')
+# Verify that you're logged in
+user = api.whoami()
+print(user)
 # Load model and tokenizer
 model_name = "meta-llama/Llama-2-7b-chat-hf"
 print("started loading model")
@@ -14,13 +25,11 @@ model = AutoModelForCausalLM.from_pretrained(
     return_dict=True,
     torch_dtype=torch.float16,
     revision="main",  # Or the desired revision
-    token=api_token  # Provide the API token here
 )
 
 tokenizer = AutoTokenizer.from_pretrained(
     model_name,
     revision="main",  # Or the desired revision
-    auth_token=api_token  # Provide the API token here
 )
 
 print("loaded model")