alexkueck commited on
Commit
a2bd8c7
1 Parent(s): 3109aba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -65,11 +65,11 @@ DOCS_DIR = "chroma/kkg"
65
  # Beim Start der Anwendung - Vektorstore laden!!!!
66
  ###########################################
67
  print("Vektorstore laden.........................")
68
- vectorstore = load_vectorstore()
69
- if vectorstore is None:
70
  print("Vektorstore neu.......................")
71
  # Initialisieren Sie den Vektorstore bei Bedarf
72
- vectorstore = create_vectorstore()
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