Shreyas094 commited on
Commit
111c87a
·
verified ·
1 Parent(s): 4892e48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ class Agent1:
49
 
50
  return questions
51
 
52
- def process(self, user_input: str) -> Dict[str, List[Dict[str, str]]]:
53
  queries = self.rephrase_and_split(user_input)
54
  print("Identified queries:", queries)
55
  results = {}
@@ -217,7 +217,7 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
217
 
218
  model = get_model(temperature, top_p, repetition_penalty)
219
  embed = get_embeddings()
220
- agent1 = Agent1(model)
221
 
222
  if os.path.exists("faiss_database"):
223
  database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
 
49
 
50
  return questions
51
 
52
+ def process(self, user_input: str) -> tuple[List[str], Dict[str, List[Dict[str, str]]]]:
53
  queries = self.rephrase_and_split(user_input)
54
  print("Identified queries:", queries)
55
  results = {}
 
217
 
218
  model = get_model(temperature, top_p, repetition_penalty)
219
  embed = get_embeddings()
220
+ agent1 = Agent1() # Create Agent1 without passing a model
221
 
222
  if os.path.exists("faiss_database"):
223
  database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)