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":
|
82 |
-
"repetition_penalty":
|
83 |
}
|
84 |
|
85 |
if not user_request.strip():
|
|
|
76 |
return str(e)
|
77 |
|
78 |
generation_config = {
|
79 |
+
"max_new_tokens": 1024, # Giới hạn số token tạo mới
|
80 |
"do_sample": False,
|
81 |
+
"num_beams": 5,
|
82 |
+
"repetition_penalty": 3.5
|
83 |
}
|
84 |
|
85 |
if not user_request.strip():
|