Update src/model.py
Browse files- 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
|