alexkueck commited on
Commit
25fd7a1
1 Parent(s): 3685ab4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -44,7 +44,7 @@ CHROMA_PDF = './chroma/kkg/pdf'
44
  CHROMA_WORD = './chroma/kkg/word'
45
  CHROMA_EXCEL = './chroma/kkg/excel'
46
  #HuggingFace Model name--------------------------------
47
- MODEL_NAME_HF = "mistralai/Mixtral-8x7B-Instruct-v0.1"
48
 
49
  #HuggingFace Reop ID--------------------------------
50
  #repo_id = "meta-llama/Llama-2-13b-chat-hf"
@@ -159,8 +159,6 @@ def add_text2(chatbot, prompt):
159
  chatbot = chatbot + [("", "Prompt fehlt!")]
160
  else:
161
  chatbot = chatbot + [(prompt, None)]
162
- print("chatbot nach add_text............")
163
- print(chatbot)
164
  return chatbot, prompt, ""
165
 
166
 
@@ -214,6 +212,11 @@ def generate_text (prompt, chatbot, history, vektordatenbank, retriever, top_p=0
214
  top_p=top_p,
215
  repetition_penalty=repetition_penalty
216
  )
 
 
 
 
 
217
 
218
  #Prompt an history anhängen und einen Text daraus machen
219
  history_text_und_prompt = generate_prompt_with_history(prompt, history)
@@ -271,7 +274,7 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
271
  #summary += " ".join(['Dokument: ' + str(doc['titel']) + ' Seite: ' + str(doc['seite']) + '\nAuschnitt: ' + str(doc["content"]) for doc in results['relevant_docs']])
272
  summary += " ".join([
273
  '<b>\nDokument: </b> <span style="color: #BB70FC;">' + str(doc['titel']) + '</span> '
274
- '<span style="color: red;"> (Seite:</span> ' + str(doc['seite']) + ')<br>'
275
  '<b>Auschnitt:</b> ' + str(doc["content"]) + '\n'
276
  for doc in results['relevant_docs']
277
  ])
 
44
  CHROMA_WORD = './chroma/kkg/word'
45
  CHROMA_EXCEL = './chroma/kkg/excel'
46
  #HuggingFace Model name--------------------------------
47
+ MODEL_NAME_HF = "HuggingFaceH4/zephyr-7b-alpha" #"mistralai/Mixtral-8x7B-Instruct-v0.1"
48
 
49
  #HuggingFace Reop ID--------------------------------
50
  #repo_id = "meta-llama/Llama-2-13b-chat-hf"
 
159
  chatbot = chatbot + [("", "Prompt fehlt!")]
160
  else:
161
  chatbot = chatbot + [(prompt, None)]
 
 
162
  return chatbot, prompt, ""
163
 
164
 
 
212
  top_p=top_p,
213
  repetition_penalty=repetition_penalty
214
  )
215
+ #######################################################
216
+ #Alternativ, wenn repro_id gegeben:
217
+ # Verwenden Sie die InferenceApi von huggingface_hub
218
+ llm = InferenceApi(repo_id, token=hf_token)
219
+
220
 
221
  #Prompt an history anhängen und einen Text daraus machen
222
  history_text_und_prompt = generate_prompt_with_history(prompt, history)
 
274
  #summary += " ".join(['Dokument: ' + str(doc['titel']) + ' Seite: ' + str(doc['seite']) + '\nAuschnitt: ' + str(doc["content"]) for doc in results['relevant_docs']])
275
  summary += " ".join([
276
  '<b>\nDokument: </b> <span style="color: #BB70FC;">' + str(doc['titel']) + '</span> '
277
+ ' (<b>Seite:</b></span> <span style="color: red;">' + str(doc['seite']) + ')<br>'
278
  '<b>Auschnitt:</b> ' + str(doc["content"]) + '\n'
279
  for doc in results['relevant_docs']
280
  ])