sitammeur commited on
Commit
bc5da26
1 Parent(s): a8d41d6

Update src/model.py

Browse files
Files changed (1) hide show
  1. src/model.py +3 -4
src/model.py CHANGED
@@ -21,14 +21,13 @@ def run_example(task_prompt, image, text_input=None):
21
  Runs an example using the given task prompt and image.
22
 
23
  Args:
24
- task_prompt (str): The task prompt for the example.
25
- image (PIL.Image.Image): The image to be processed.
26
- text_input (str, optional): Additional text input to be appended to the task prompt. Defaults to None.
27
 
28
  Returns:
29
  str: The parsed answer generated by the model.
30
  """
31
-
32
  # If there is no text input, use the task prompt as the prompt
33
  if text_input is None:
34
  prompt = task_prompt
 
21
  Runs an example using the given task prompt and image.
22
 
23
  Args:
24
+ - task_prompt (str): The task prompt for the example.
25
+ - image (PIL.Image.Image): The image to be processed.
26
+ - text_input (str, optional): Additional text input to be appended to the task prompt. Defaults to None.
27
 
28
  Returns:
29
  str: The parsed answer generated by the model.
30
  """
 
31
  # If there is no text input, use the task prompt as the prompt
32
  if text_input is None:
33
  prompt = task_prompt