Embrw commited on
Commit
41cb1c4
·
verified ·
1 Parent(s): 7856f25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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": 5,
82
- "repetition_penalty": 3.5
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")