AarushiGupta28 commited on
Commit
9fddb39
·
verified ·
1 Parent(s): bd65d5d

Update app_server.py

Browse files
Files changed (1) hide show
  1. 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
- #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,7 +71,7 @@ def fetch_data_from_api(url):
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,
 
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,