rishisim commited on
Commit
907b68f
1 Parent(s): 7fcf7c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -7,11 +7,15 @@ from langchain.vectorstores import FAISS
7
  from langchain.prompts import PromptTemplate
8
  from langchain.chains import RetrievalQA
9
  import warnings
 
10
 
11
 
12
  # from transformers import AutoModelForCausalLM, AutoTokenizer
13
  from transformers import pipeline
14
 
 
 
 
15
  llm = pipeline("text-generation", model = "meta-llama/Meta-Llama-3-70B-Instruct")
16
 
17
  # llm = pipeline("text-generation", model="mistralai/Mixtral-8x22B-Instruct-v0.1")
 
7
  from langchain.prompts import PromptTemplate
8
  from langchain.chains import RetrievalQA
9
  import warnings
10
+ from huggingface_hub import login
11
 
12
 
13
  # from transformers import AutoModelForCausalLM, AutoTokenizer
14
  from transformers import pipeline
15
 
16
+ login(token="your_huggingface_api_token")
17
+
18
+
19
  llm = pipeline("text-generation", model = "meta-llama/Meta-Llama-3-70B-Instruct")
20
 
21
  # llm = pipeline("text-generation", model="mistralai/Mixtral-8x22B-Instruct-v0.1")