alexkueck commited on
Commit
4fced9d
1 Parent(s): 5068dc1

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -2
utils.py CHANGED
@@ -407,8 +407,7 @@ def rag_chain(llm, prompt, retriever):
407
 
408
  #3. Alternative: mit pipeline
409
  messages = [{"role": "user", "content": input_text},]
410
- pipe = pipeline("text-generation", model= llm, trust_remote_code=True)
411
- answer = pipe(messages)
412
 
413
  # Erstelle das Ergebnis-Dictionary
414
  result = {
 
407
 
408
  #3. Alternative: mit pipeline
409
  messages = [{"role": "user", "content": input_text},]
410
+ answer = llm(messages)
 
411
 
412
  # Erstelle das Ergebnis-Dictionary
413
  result = {