alexkueck commited on
Commit
0656780
1 Parent(s): c2d46cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def predict(text,
90
  """
91
  ##########################################################################
92
  #Prompt ist erzeugt, nun mit pipeline eine Antwort von der KI bekommen!
93
- pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
94
  bot_message = pipe(prompt)
95
 
96
  #chatbot - history erweitern und an chatbotGr zurückschicken
 
90
  """
91
  ##########################################################################
92
  #Prompt ist erzeugt, nun mit pipeline eine Antwort von der KI bekommen!
93
+ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0, do_sample=True, top_p=0.95)
94
  bot_message = pipe(prompt)
95
 
96
  #chatbot - history erweitern und an chatbotGr zurückschicken