Update app.py
Browse files
app.py
CHANGED
@@ -78,8 +78,8 @@ def process_image(image, user_request):
|
|
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":
|
82 |
-
"repetition_penalty":
|
83 |
}
|
84 |
|
85 |
if not user_request.strip():
|
@@ -109,4 +109,4 @@ iface = gr.Interface(
|
|
109 |
)
|
110 |
|
111 |
if __name__ == "__main__":
|
112 |
-
iface.launch()
|
|
|
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": 3,
|
82 |
+
"repetition_penalty": 2.5
|
83 |
}
|
84 |
|
85 |
if not user_request.strip():
|
|
|
109 |
)
|
110 |
|
111 |
if __name__ == "__main__":
|
112 |
+
iface.launch("=True")
|