Nguyễn Hoàng Ân
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -76,10 +76,10 @@ def process_image(image, user_request):
|
|
| 76 |
return str(e)
|
| 77 |
|
| 78 |
generation_config = {
|
| 79 |
-
"max_new_tokens":
|
| 80 |
"do_sample": False,
|
| 81 |
"num_beams": 3,
|
| 82 |
-
"repetition_penalty": 2.
|
| 83 |
}
|
| 84 |
|
| 85 |
if not user_request.strip():
|
|
|
|
| 76 |
return str(e)
|
| 77 |
|
| 78 |
generation_config = {
|
| 79 |
+
"max_new_tokens": 568, # Giới hạn số token tạo mới
|
| 80 |
"do_sample": False,
|
| 81 |
"num_beams": 3,
|
| 82 |
+
"repetition_penalty": 2.0
|
| 83 |
}
|
| 84 |
|
| 85 |
if not user_request.strip():
|