Update app.py
Browse files
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=
|
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)
|