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 = {
|