Update utils.py
Browse files
utils.py
CHANGED
|
@@ -405,8 +405,10 @@ def rag_chain(llm, prompt, retriever):
|
|
| 405 |
#answer = query(llm, {"inputs": input_text,})
|
| 406 |
|
| 407 |
#3. Alternative: mit pipeline
|
| 408 |
-
|
| 409 |
-
|
|
|
|
|
|
|
| 410 |
|
| 411 |
# Erstelle das Ergebnis-Dictionary
|
| 412 |
result = {
|
|
|
|
| 405 |
#answer = query(llm, {"inputs": input_text,})
|
| 406 |
|
| 407 |
#3. Alternative: mit pipeline
|
| 408 |
+
#für text-generation:
|
| 409 |
+
#messages = [{"role": "user", "content": input_text},]
|
| 410 |
+
#für summarizatiuon
|
| 411 |
+
answer = llm(input_text, max_length=512, min_length=150, do_sample=False)
|
| 412 |
|
| 413 |
# Erstelle das Ergebnis-Dictionary
|
| 414 |
result = {
|