Spaces:
Running
on
Zero
Running
on
Zero
Update src/app/response.py
Browse files- src/app/response.py +1 -1
src/app/response.py
CHANGED
@@ -25,7 +25,7 @@ model, processor = load_model_and_processor(model_name, device)
|
|
25 |
|
26 |
@spaces.GPU
|
27 |
def caption_image(
|
28 |
-
image: PIL.Image.Image, max_new_tokens: int,
|
29 |
) -> str:
|
30 |
"""
|
31 |
Generates a caption based on the given image using the model.
|
|
|
25 |
|
26 |
@spaces.GPU
|
27 |
def caption_image(
|
28 |
+
image: PIL.Image.Image, max_new_tokens: int, language: str, sampling: bool
|
29 |
) -> str:
|
30 |
"""
|
31 |
Generates a caption based on the given image using the model.
|