Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -61,8 +61,8 @@ def get_qwen_op(image_file, model, processor):
|
|
61 |
inputs = {k: v.to(torch.float32) if torch.is_floating_point(v) else v for k, v in inputs.items()}
|
62 |
|
63 |
generation_config = {
|
64 |
-
"max_new_tokens":
|
65 |
-
"do_sample":
|
66 |
"top_k": 20,
|
67 |
"top_p": 0.90,
|
68 |
"temperature": 0.4,
|
|
|
61 |
inputs = {k: v.to(torch.float32) if torch.is_floating_point(v) else v for k, v in inputs.items()}
|
62 |
|
63 |
generation_config = {
|
64 |
+
"max_new_tokens": 10000,
|
65 |
+
"do_sample": False,
|
66 |
"top_k": 20,
|
67 |
"top_p": 0.90,
|
68 |
"temperature": 0.4,
|