sitammeur commited on
Commit
b966f2d
·
verified ·
1 Parent(s): 81e3667

Update src/app/response.py

Browse files
Files changed (1) hide show
  1. 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, sampling: bool, language: str
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.