Update app.py
Browse files
app.py
CHANGED
@@ -238,13 +238,11 @@ def generate_text (prompt, chatbot, history, vektordatenbank, retriever, top_p=0
|
|
238 |
llm = HuggingFaceEndpoint(
|
239 |
endpoint_url=f"https://api-inference.huggingface.co/models/{MODEL_NAME_HF}",
|
240 |
api_key=hf_token,
|
241 |
-
|
242 |
-
|
243 |
-
|
244 |
-
|
245 |
-
|
246 |
-
"repetition_penalty": repetition_penalty
|
247 |
-
}
|
248 |
)
|
249 |
result = rag_chain(llm, history_text_und_prompt, retriever)
|
250 |
|
|
|
238 |
llm = HuggingFaceEndpoint(
|
239 |
endpoint_url=f"https://api-inference.huggingface.co/models/{MODEL_NAME_HF}",
|
240 |
api_key=hf_token,
|
241 |
+
temperature=0.5,
|
242 |
+
max_length=1024,
|
243 |
+
top_k=top_k,
|
244 |
+
top_p=top_p,
|
245 |
+
repetition_penalty=repetition_penalty
|
|
|
|
|
246 |
)
|
247 |
result = rag_chain(llm, history_text_und_prompt, retriever)
|
248 |
|