SallySims commited on
Commit
7d9617d
·
verified ·
1 Parent(s): d0d8e5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -10,6 +10,11 @@ st.set_page_config(page_title="AnthroBot", page_icon="🤖", layout="centered")
10
 
11
  # Load model & tokenizer
12
  @st.cache_resource
 
 
 
 
 
13
  def load_model():
14
  peft_config = PeftConfig.from_pretrained("SallySims/AnthroBot_Model_Lora")
15
  base_model = AutoModelForCausalLM.from_pretrained(
 
10
 
11
  # Load model & tokenizer
12
  @st.cache_resource
13
+ from huggingface_hub import login
14
+
15
+ # Log in using the token
16
+ login(token=os.getenv("HUGGINGFACEHUB_TOKEN"))
17
+
18
  def load_model():
19
  peft_config = PeftConfig.from_pretrained("SallySims/AnthroBot_Model_Lora")
20
  base_model = AutoModelForCausalLM.from_pretrained(