Updated login procedure to hopefully make it work

app_inference.py

@@ -3,7 +3,7 @@ import streamlit as st
 import transformers, torch
 import json, os
 
-from huggingface_hub import HfApi
+from huggingface_hub import login
 
 # CONSTANTS
 MAX_NEW_TOKENS = 256
@@ -18,8 +18,8 @@ submit = st.button("Submit")
 
 # MODEL AREA
 # Use the token to authenticate
-token = os.environ.get("HF_TOKEN_READ")
-api = HfApi(token=token)
+token = os.getenv("HF_TOKEN_READ")
+api = login(token=token)
 model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
 @st.cache_resource

examples/tests/hf_login.py

@@ -1,4 +1,4 @@
-# from huggingface_hub import login
+from huggingface_hub import login
 # login()
 
 import transformers, torch
@@ -12,7 +12,7 @@ token = os.environ.get("HF_TOKEN_READ")
 print(token)
 
 # Use the token to authenticate
-api = HfApi(token=token)
+api = login(token=token)
 model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
 pipeline = transformers.pipeline(