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)
|