Update app.py
Browse files
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;">
|
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 |
])
|