sagarkariya commited on
Commit
783fc68
·
verified ·
1 Parent(s): 43185fb

added code for hf Token in app.ppy

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -6,6 +6,13 @@ import numpy as np
6
  from langchain.embeddings import HuggingFaceEmbeddings
7
  from langchain.vectorstores import FAISS
8
 
 
 
 
 
 
 
 
9
  # 🎯 Load CodeLlama 3B Model
10
  model_name = "codellama/CodeLlama-3b"
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
6
  from langchain.embeddings import HuggingFaceEmbeddings
7
  from langchain.vectorstores import FAISS
8
 
9
+ hf_token = os.getenv("PROJECT_TEST_TOKEN")
10
+ if hf_token:
11
+ login(hf_token)
12
+ else:
13
+ st.error("Token not found")
14
+ st.stop()
15
+
16
  # 🎯 Load CodeLlama 3B Model
17
  model_name = "codellama/CodeLlama-3b"
18
  tokenizer = AutoTokenizer.from_pretrained(model_name)