Update app.py
Browse files
app.py
CHANGED
@@ -65,11 +65,11 @@ DOCS_DIR = "chroma/kkg"
|
|
65 |
# Beim Start der Anwendung - Vektorstore laden!!!!
|
66 |
###########################################
|
67 |
print("Vektorstore laden.........................")
|
68 |
-
|
69 |
-
if
|
70 |
print("Vektorstore neu.......................")
|
71 |
# Initialisieren Sie den Vektorstore bei Bedarf
|
72 |
-
|
73 |
|
74 |
###########################################
|
75 |
# Alternativen, um HF Modelle in der rAG Chain einzusetzen
|
@@ -279,6 +279,8 @@ def generate_text (prompt, chatbot, history, retriever, top_p=0.6, temperature=0
|
|
279 |
##############################################################
|
280 |
#Eingaben der GUI verarbeiten
|
281 |
def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_docs=4, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,top_k=5, validate=False):
|
|
|
|
|
282 |
#nur wenn man sich validiert hat, kann die Anwendung los legen
|
283 |
if (validate and not prompt_in == "" and not prompt_in == None):
|
284 |
# Vektorstore initialisieren
|
|
|
65 |
# Beim Start der Anwendung - Vektorstore laden!!!!
|
66 |
###########################################
|
67 |
print("Vektorstore laden.........................")
|
68 |
+
vektordatenbank = load_vectorstore()
|
69 |
+
if vektordatenbank is None:
|
70 |
print("Vektorstore neu.......................")
|
71 |
# Initialisieren Sie den Vektorstore bei Bedarf
|
72 |
+
vektordatenbank = create_vectorstore()
|
73 |
|
74 |
###########################################
|
75 |
# Alternativen, um HF Modelle in der rAG Chain einzusetzen
|
|
|
279 |
##############################################################
|
280 |
#Eingaben der GUI verarbeiten
|
281 |
def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_docs=4, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,top_k=5, validate=False):
|
282 |
+
global vektordatenbank
|
283 |
+
|
284 |
#nur wenn man sich validiert hat, kann die Anwendung los legen
|
285 |
if (validate and not prompt_in == "" and not prompt_in == None):
|
286 |
# Vektorstore initialisieren
|