Spaces:
Sleeping
Sleeping
Chengxb888
commited on
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']}
|