Chandranshu Jain commited on
Commit
c18ef09
1 Parent(s): 6977def

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -13,8 +13,7 @@ from langchain.prompts import PromptTemplate
13
  from langchain_community.document_loaders import PyPDFLoader
14
  from langchain_chroma import Chroma
15
  from langchain_community.vectorstores import Chroma
16
- # Use a pipeline as a high-level helper
17
- from transformers import pipeline
18
 
19
 
20
 
@@ -64,7 +63,7 @@ def text_splitter(text):
64
 
65
  #GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
66
  #COHERE_API_KEY = os.getenv("COHERE_API_KEY")
67
- HUGGING_FACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
68
 
69
  def get_conversational_chain():
70
  prompt_template = """
@@ -83,7 +82,7 @@ def get_conversational_chain():
83
  repo_id ='google/gemma-1.1-2b-it'
84
  #repo_id='meta-llama/Meta-Llama-3-70B'
85
  llm = HuggingFaceEndpoint(
86
- repo_id=repo_id, max_length=512, temperature=0.3, token=HUGGING_FACE_API_KEY)
87
  #tokenizer = AutoTokenizer.from_pretrained("google/gemma-1.1-2b-it")
88
  #llm = AutoModelForCausalLM.from_pretrained("google/gemma-1.1-2b-it")
89
 
 
13
  from langchain_community.document_loaders import PyPDFLoader
14
  from langchain_chroma import Chroma
15
  from langchain_community.vectorstores import Chroma
16
+
 
17
 
18
 
19
 
 
63
 
64
  #GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
65
  #COHERE_API_KEY = os.getenv("COHERE_API_KEY")
66
+ HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
67
 
68
  def get_conversational_chain():
69
  prompt_template = """
 
82
  repo_id ='google/gemma-1.1-2b-it'
83
  #repo_id='meta-llama/Meta-Llama-3-70B'
84
  llm = HuggingFaceEndpoint(
85
+ repo_id=repo_id, max_length=512, temperature=0.3, token=HUGGINGFACE_API_KEY)
86
  #tokenizer = AutoTokenizer.from_pretrained("google/gemma-1.1-2b-it")
87
  #llm = AutoModelForCausalLM.from_pretrained("google/gemma-1.1-2b-it")
88