Spaces:
Running
Running
Chandranshu Jain
commited on
Update app.py
Browse files
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 |
-
|
| 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 |
-
|
| 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=
|
| 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 |
|