Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|