Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ terminators = [
|
|
| 37 |
]
|
| 38 |
|
| 39 |
generation_params = {
|
| 40 |
-
'max_new_tokens':
|
| 41 |
'use_cache': True,
|
| 42 |
'do_sample': True,
|
| 43 |
'temperature': 0.7,
|
|
@@ -79,6 +79,7 @@ def inference(query):
|
|
| 79 |
|
| 80 |
4. get input language and translate it inti target language.
|
| 81 |
- return only translation. without extra explaination and comments.
|
|
|
|
| 82 |
|
| 83 |
"""},
|
| 84 |
{"role": "user", "content": f"{query}"},
|
|
|
|
| 37 |
]
|
| 38 |
|
| 39 |
generation_params = {
|
| 40 |
+
'max_new_tokens': 2000,
|
| 41 |
'use_cache': True,
|
| 42 |
'do_sample': True,
|
| 43 |
'temperature': 0.7,
|
|
|
|
| 79 |
|
| 80 |
4. get input language and translate it inti target language.
|
| 81 |
- return only translation. without extra explaination and comments.
|
| 82 |
+
- do not return extra text.
|
| 83 |
|
| 84 |
"""},
|
| 85 |
{"role": "user", "content": f"{query}"},
|