Thiloid commited on
Commit
6bb1bf8
·
verified ·
1 Parent(s): a8bfd11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,7 +55,7 @@ def response(prompt, history, selected_option, temperature=0.9, max_new_tokens=5
55
  addon = ""
56
  results=collection.query(
57
  query_texts=[search_prompt],
58
- n_results=60
59
  )
60
 
61
  dists = ["<br><small>(relevance: " + str(round((1 - d) * 100) / 100) + ";" for d in results['distances'][0]]
@@ -65,7 +65,7 @@ def response(prompt, history, selected_option, temperature=0.9, max_new_tokens=5
65
  print(results)
66
  if len(results) > 1:
67
  addon = "Bitte berücksichtige bei deiner Antwort ausschießlich folgende Auszüge aus unserer Wahlhilfe Datenbank, sofern sie für die Antwort erforderlich sind. Beantworte die Frage knapp und präzise. Ignoriere unpassende Datenbank-Auszüge OHNE sie zu kommentieren, zu erwähnen oder aufzulisten:\n" + "\n".join(results)
68
- system = "Du heißt Avah und bist ein deutschsprachiges KI-basiertes Wahlhilfe Beratungssystem zum Wahl-O-Maten für die Bundestagswahl 2021, an welchem insgesamt 38 Parteien teilgenommen haben. Du nimmst selber keine eigene politische Postion ein!" + addon + "\n\nUser-Anliegen:"
69
 
70
  formatted_prompt = format_prompt(system + "\n" + selected_option + "\n" + prompt, history)
71
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
 
55
  addon = ""
56
  results=collection.query(
57
  query_texts=[search_prompt],
58
+ n_results=10
59
  )
60
 
61
  dists = ["<br><small>(relevance: " + str(round((1 - d) * 100) / 100) + ";" for d in results['distances'][0]]
 
65
  print(results)
66
  if len(results) > 1:
67
  addon = "Bitte berücksichtige bei deiner Antwort ausschießlich folgende Auszüge aus unserer Wahlhilfe Datenbank, sofern sie für die Antwort erforderlich sind. Beantworte die Frage knapp und präzise. Ignoriere unpassende Datenbank-Auszüge OHNE sie zu kommentieren, zu erwähnen oder aufzulisten:\n" + "\n".join(results)
68
+ system = "Du heißt Avah und bist ein deutschsprachiges KI-basiertes Wahlhilfe Beratungssystem zum Wahl-O-Maten für die Bundestagswahl 2021, an welchem insgesamt 38 Parteien teilgenommen haben. Du beinhaltest aber nur Informationen zu den 6 Hauptparteien. Du nimmst selber keine eigene politische Postion ein!" + addon + "\n\nUser-Anliegen:"
69
 
70
  formatted_prompt = format_prompt(system + "\n" + selected_option + "\n" + prompt, history)
71
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)