alexkueck commited on
Commit
d83ad17
1 Parent(s): 0a32c4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,7 +55,7 @@ CHROMA_PDF = './chroma/kkg/pdf'
55
  CHROMA_WORD = './chroma/kkg/word'
56
  CHROMA_EXCEL = './chroma/kkg/excel'
57
  #HuggingFace Model name--------------------------------
58
- MODEL_NAME_HF = "t5-small" #"meta-llama/Meta-Llama-3-8B-Instruct" #"mistralai/Mistral-7B-Instruct-v0.3" #"microsoft/Phi-3-mini-4k-instruct" #"HuggingFaceH4/zephyr-7b-alpha"
59
 
60
  #HuggingFace Reop ID--------------------------------
61
  #repo_id = "meta-llama/Llama-2-13b-chat-hf"
@@ -297,7 +297,7 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
297
  summary = "<b>Zusammenfassung: </b>\n" + str(results['answer']) + "\n\n<b>Auszüge dazu: </b>"
298
  summary += " ".join([
299
  '<div><b>Dokument: </b> <span style="color: #BB70FC;">' + str(doc['titel']) + '</span> '
300
- '(<b>Seite:</b></span> <span style="color: red;">' + str(doc['seite']) + '</span>)<br>'
301
  '<span><b>Auschnitt:</b> ' + str(doc["content"]) + '</span></div>'
302
  '<div><span><b>Link: </b><span style="color: #BB70FC;"><a href="' + str(doc['download_link']) + '" target="_blank">' + str(doc['titel']) + '</a></span></div><br>'
303
  for doc in results['relevant_docs']])
 
55
  CHROMA_WORD = './chroma/kkg/word'
56
  CHROMA_EXCEL = './chroma/kkg/excel'
57
  #HuggingFace Model name--------------------------------
58
+ MODEL_NAME_HF = 'all-MiniLM-L6-v2' #"t5-small" #"meta-llama/Meta-Llama-3-8B-Instruct" #"mistralai/Mistral-7B-Instruct-v0.3" #"microsoft/Phi-3-mini-4k-instruct" #"HuggingFaceH4/zephyr-7b-alpha"
59
 
60
  #HuggingFace Reop ID--------------------------------
61
  #repo_id = "meta-llama/Llama-2-13b-chat-hf"
 
297
  summary = "<b>Zusammenfassung: </b>\n" + str(results['answer']) + "\n\n<b>Auszüge dazu: </b>"
298
  summary += " ".join([
299
  '<div><b>Dokument: </b> <span style="color: #BB70FC;">' + str(doc['titel']) + '</span> '
300
+ '(<b>Seite:</span> <span style="color: red;">' + str(doc['seite']) + '</b></span>)<br>'
301
  '<span><b>Auschnitt:</b> ' + str(doc["content"]) + '</span></div>'
302
  '<div><span><b>Link: </b><span style="color: #BB70FC;"><a href="' + str(doc['download_link']) + '" target="_blank">' + str(doc['titel']) + '</a></span></div><br>'
303
  for doc in results['relevant_docs']])