Spaces:
Sleeping
Sleeping
Commit
·
8b0cb96
1
Parent(s):
40e6fe6
collection name
Browse files
app.py
CHANGED
|
@@ -173,12 +173,12 @@ def respond(
|
|
| 173 |
<|im_start|>assistant"""
|
| 174 |
refined_context = generator[1](system_message, do_sample=True, padding=True, truncation=True, top_p=0.95, max_new_tokens=150)
|
| 175 |
# Retrieve relevant context from Qdrant
|
|
|
|
| 176 |
if "rezept" in query.lower() or "gericht" in query.lower():
|
| 177 |
collection_name = "recipes"
|
| 178 |
elif "bestellung" in query.lower() or "order" in query.lower():
|
| 179 |
collection_name = "products"
|
| 180 |
-
|
| 181 |
-
colleciton_name = "products"
|
| 182 |
|
| 183 |
context = search_qdrant_with_context(query + " " + refined_context[0]["generated_text"].split("assistant\n").pop(), collection_name)
|
| 184 |
answer = generate_response(query, context, last_messages, max_tokens, temperature, top_p, generator[0])
|
|
|
|
| 173 |
<|im_start|>assistant"""
|
| 174 |
refined_context = generator[1](system_message, do_sample=True, padding=True, truncation=True, top_p=0.95, max_new_tokens=150)
|
| 175 |
# Retrieve relevant context from Qdrant
|
| 176 |
+
collection_name = "products"
|
| 177 |
if "rezept" in query.lower() or "gericht" in query.lower():
|
| 178 |
collection_name = "recipes"
|
| 179 |
elif "bestellung" in query.lower() or "order" in query.lower():
|
| 180 |
collection_name = "products"
|
| 181 |
+
|
|
|
|
| 182 |
|
| 183 |
context = search_qdrant_with_context(query + " " + refined_context[0]["generated_text"].split("assistant\n").pop(), collection_name)
|
| 184 |
answer = generate_response(query, context, last_messages, max_tokens, temperature, top_p, generator[0])
|