Aliayub1995 commited on
Commit
c046ba1
1 Parent(s): 89a6901

Update videollama2/__init__.py

Browse files
Files changed (1) hide show
  1. videollama2/__init__.py +1 -1
videollama2/__init__.py CHANGED
@@ -33,7 +33,7 @@ def model_init(model_path=None, **kwargs):
33
  return model, processor, tokenizer
34
 
35
 
36
- def mm_infer(url, instruct, model, tokenizer, modal='video', **kwargs):
37
  """inference api of VideoLLaMA2 for video understanding.
38
 
39
  Args:
 
33
  return model, processor, tokenizer
34
 
35
 
36
+ def mm_infer(image_or_video, instruct, model, tokenizer, modal='video', **kwargs):
37
  """inference api of VideoLLaMA2 for video understanding.
38
 
39
  Args: