Update app.py
Browse files
app.py
CHANGED
@@ -115,7 +115,7 @@ def draw_ocr_bboxes(image, prediction):
|
|
115 |
fill=color)
|
116 |
return image
|
117 |
|
118 |
-
def process_image(image, task_prompt, text_input=None, model_id='
|
119 |
image = Image.fromarray(image) # Convert NumPy array to PIL Image
|
120 |
if task_prompt == 'Caption':
|
121 |
task_prompt = '<CAPTION>'
|
@@ -251,7 +251,7 @@ with gr.Blocks(css=css) as demo:
|
|
251 |
with gr.Row():
|
252 |
with gr.Column():
|
253 |
input_img = gr.Image(label="Input Picture")
|
254 |
-
model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='
|
255 |
task_type = gr.Radio(choices=['Single task', 'Cascased task'], label='Task type selector', value='Single task')
|
256 |
task_prompt = gr.Dropdown(choices=single_task_list, label="Task Prompt", value="Caption")
|
257 |
task_type.change(fn=update_task_dropdown, inputs=task_type, outputs=task_prompt)
|
|
|
115 |
fill=color)
|
116 |
return image
|
117 |
|
118 |
+
def process_image(image, task_prompt, text_input=None, model_id='J-LAB/Florence-Idesire'):
|
119 |
image = Image.fromarray(image) # Convert NumPy array to PIL Image
|
120 |
if task_prompt == 'Caption':
|
121 |
task_prompt = '<CAPTION>'
|
|
|
251 |
with gr.Row():
|
252 |
with gr.Column():
|
253 |
input_img = gr.Image(label="Input Picture")
|
254 |
+
model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='J-LAB/Florence-Idesire')
|
255 |
task_type = gr.Radio(choices=['Single task', 'Cascased task'], label='Task type selector', value='Single task')
|
256 |
task_prompt = gr.Dropdown(choices=single_task_list, label="Task Prompt", value="Caption")
|
257 |
task_type.change(fn=update_task_dropdown, inputs=task_type, outputs=task_prompt)
|