Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|