Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,13 +34,13 @@ def say_hello(msg: str):
|
|
| 34 |
tokenizer=tokenizer,
|
| 35 |
)
|
| 36 |
print("output")
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
}
|
| 43 |
|
| 44 |
-
output = pipe(messages
|
| 45 |
print("complete")
|
| 46 |
return {"message": output[0]['generated_text']}
|
|
|
|
| 34 |
tokenizer=tokenizer,
|
| 35 |
)
|
| 36 |
print("output")
|
| 37 |
+
# generation_args = {
|
| 38 |
+
# "max_new_tokens": 500,
|
| 39 |
+
# "return_full_text": False,
|
| 40 |
+
# "temperature": 0.0,
|
| 41 |
+
# "do_sample": False,
|
| 42 |
+
# }
|
| 43 |
|
| 44 |
+
output = pipe(messages) #, **generation_args)
|
| 45 |
print("complete")
|
| 46 |
return {"message": output[0]['generated_text']}
|