mohamedyd commited on
Commit
28d38c2
·
verified ·
1 Parent(s): 8b21851

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,6 +1,13 @@
1
  import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  from peft import PeftModel
 
 
 
 
 
 
 
4
 
5
  # Load the base model and adapter
6
  @st.cache_resource
 
1
  import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  from peft import PeftModel
4
+ from dotenv import load_dotenv
5
+ load_dotenv()
6
+ from huggingface_hub import login
7
+ import os
8
+
9
+ access_token = os.environ["HUGGING_FACE_HUB_TOKEN"]
10
+ login(token=access_token)
11
 
12
  # Load the base model and adapter
13
  @st.cache_resource