Ankerkraut commited on
Commit
8b0cb96
·
1 Parent(s): 40e6fe6

collection name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- else:
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])