Spaces:
Runtime error
Runtime error
Update app_server.py
Browse files- app_server.py +2 -2
app_server.py
CHANGED
|
@@ -42,7 +42,7 @@ requests.packages.urllib3.disable_warnings(category=InsecureRequestWarning)
|
|
| 42 |
model = ChatGoogleGenerativeAI(model="gemini-pro",google_api_key=GOOGLE_API_KEY,temperature=0.3,convert_system_message_to_human=True)
|
| 43 |
embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001",google_api_key=GOOGLE_API_KEY)
|
| 44 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=10)
|
| 45 |
-
|
| 46 |
# persist_directory='/Users/vp317n/Documents/JUPYTER NOTEBOOK/db/'
|
| 47 |
# file_path = "100_names.txt"
|
| 48 |
# with open(file_path, 'r') as file:
|
|
@@ -71,7 +71,7 @@ def fetch_data_from_api(url):
|
|
| 71 |
|
| 72 |
print("\n\n\Actual answer:\n")
|
| 73 |
|
| 74 |
-
|
| 75 |
|
| 76 |
service_names_chain = RetrievalQA.from_chain_type(
|
| 77 |
model,
|
|
|
|
| 42 |
model = ChatGoogleGenerativeAI(model="gemini-pro",google_api_key=GOOGLE_API_KEY,temperature=0.3,convert_system_message_to_human=True)
|
| 43 |
embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001",google_api_key=GOOGLE_API_KEY)
|
| 44 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=10)
|
| 45 |
+
location = '/Users/ag396s/Documents/UI/'
|
| 46 |
# persist_directory='/Users/vp317n/Documents/JUPYTER NOTEBOOK/db/'
|
| 47 |
# file_path = "100_names.txt"
|
| 48 |
# with open(file_path, 'r') as file:
|
|
|
|
| 71 |
|
| 72 |
print("\n\n\Actual answer:\n")
|
| 73 |
|
| 74 |
+
service_names = Chroma(persist_directory=location,embedding_function=embeddings)
|
| 75 |
|
| 76 |
service_names_chain = RetrievalQA.from_chain_type(
|
| 77 |
model,
|