Spaces:
Sleeping
Sleeping
vidhanm
commited on
Commit
·
f2f290e
1
Parent(s):
08a37b7
reducing tokens to 20
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def call_generate_script(image_path: str, prompt_text: str) -> str:
|
|
| 36 |
"--image", image_path,
|
| 37 |
"--prompt", prompt_text,
|
| 38 |
"--generations", "1", # Get one generation for the UI
|
| 39 |
-
"--max_new_tokens", "
|
| 40 |
# --device is handled by generate.py internally
|
| 41 |
]
|
| 42 |
|
|
|
|
| 36 |
"--image", image_path,
|
| 37 |
"--prompt", prompt_text,
|
| 38 |
"--generations", "1", # Get one generation for the UI
|
| 39 |
+
"--max_new_tokens", "20" # Adjust as needed
|
| 40 |
# --device is handled by generate.py internally
|
| 41 |
]
|
| 42 |
|