alexkueck commited on
Commit
3840048
1 Parent(s): 5c58cc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -190,7 +190,7 @@ def generate_text (prompt, chatbot, history, vektordatenbank, retriever, top_p=0
190
  #llm = HuggingFaceHub(repo_id=repo_id, model_kwargs=model_kwargs)
191
  #llm = HuggingFaceChain(model=MODEL_NAME_HF, model_kwargs={"temperature": 0.5, "max_length": 128})
192
  # Erstelle eine Pipeline mit den gewünschten Parametern
193
- pipe = pipeline("text-generation", model=model_name, model_kwargs=model_kwargs)
194
 
195
  # Erstelle eine HuggingFacePipeline-Kette
196
  llm = HuggingFacePipeline(pipeline=pipe)
 
190
  #llm = HuggingFaceHub(repo_id=repo_id, model_kwargs=model_kwargs)
191
  #llm = HuggingFaceChain(model=MODEL_NAME_HF, model_kwargs={"temperature": 0.5, "max_length": 128})
192
  # Erstelle eine Pipeline mit den gewünschten Parametern
193
+ pipe = pipeline("text-generation", model=MODEL_NAME_HF , model_kwargs=model_kwargs)
194
 
195
  # Erstelle eine HuggingFacePipeline-Kette
196
  llm = HuggingFacePipeline(pipeline=pipe)