Update app.py

app.py

@@ -1,7 +1,7 @@
-import spaces
 import gradio as gr
 from transformers import AutoProcessor, AutoModelForCausalLM
 # import peft
+import spaces
 
 import requests
 import copy
@@ -20,14 +20,11 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
 models = {
     'microsoft/Florence-2-large-ft': AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-large-ft', trust_remote_code=True).to("cuda").eval(),
     'dwb2023/florence2-large-bccd-base-ft': AutoModelForCausalLM.from_pretrained('dwb2023/florence2-large-bccd-base-ft', trust_remote_code=True).to("cuda").eval(),
-    'dwb2023/florence2-large-liver-disease-ft': AutoModelForCausalLM.from_pretrained('dwb2023/florence2-large-liver-disease-ft', trust_remote_code=True).to("cuda").eval(),
-
 }
 
 processors = {
     'microsoft/Florence-2-large-ft': AutoProcessor.from_pretrained('microsoft/Florence-2-large-ft', trust_remote_code=True),
-    'dwb2023/florence2-large-bccd-base-ft': AutoProcessor.from_pretrained('microsoft/Florence-2-large-ft', trust_remote_code=True),
-    'dwb2023/florence2-large-liver-disease-ft': AutoProcessor.from_pretrained('microsoft/Florence-2-large-ft', trust_remote_code=True),
+    'dwb2023/florence2-large-bccd-base-ft': AutoProcessor.from_pretrained('microsoft/Florence-2-large-ft', trust_remote_code=True),  
 }
 
 colormap = ['blue','orange','green','purple','brown','pink','gray','olive','cyan','red',
@@ -40,7 +37,7 @@ def fig_to_pil(fig):
     return Image.open(buf)
 
 @spaces.GPU
-def run_example(task_prompt, image, text_input=None, model_id='dwb2023/florence2-large-liver-disease-ft'):
+def run_example(task_prompt, image, text_input=None, model_id='dwb2023/florence2-large-bccd-base-ft'):
     model = models[model_id]
     processor = processors[model_id]
     if text_input is None:
@@ -118,7 +115,7 @@ def draw_ocr_bboxes(image, prediction):
                   fill=color)
     return image
 
-def process_image(image, task_prompt, text_input=None, model_id='dwb2023/florence2-large-liver-disease-ft'):
+def process_image(image, task_prompt, text_input=None, model_id='dwb2023/florence2-large-bccd-base-ft'):
     image = Image.fromarray(image)  # Convert NumPy array to PIL Image
     if task_prompt == 'Object Detection':
         task_prompt = '<OD>'
@@ -145,7 +142,7 @@ with gr.Blocks(theme="sudeepshouche/minimalist") as demo:
         with gr.Row():
             with gr.Column():
                 input_img = gr.Image(label="Input Picture")
-                model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='dwb2023/florence2-large-liver-disease-ft')
+                model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='dwb2023/florence2-large-bccd-base-ft')
                 task_prompt = gr.Dropdown(choices=single_task_list, label="Task Prompt", value="Object Detection")
                 text_input = gr.Textbox(label="Text Input", placeholder="Not used for Florence-2 Object Detection")
                 submit_btn = gr.Button(value="Submit")
@@ -161,10 +158,8 @@ with gr.Blocks(theme="sudeepshouche/minimalist") as demo:
                 ["examples/bccd-test/BloodImage_00090_jpg.rf.7e3d419774b20ef93d4ec6c4be8f64df.jpg", 'Object Detection'],
                 ["examples/bccd-test/BloodImage_00099_jpg.rf.0a65e56401cdd71253e7bc04917c3558.jpg", 'Object Detection'],
                 ["examples/bccd-test/BloodImage_00112_jpg.rf.6b8d185de08e65c6d765c824bb76ec68.jpg", 'Object Detection'],
-                ["examples/liver-test/15_239_101_64_47_jpg.rf.c23cc3f6ba46896ce9c2cf3ee242bbaf.jpg", 'Object Detection'],
-                ["examples/liver-test/15_242_208_18_9_jpg.rf.b1ae586b5c888d8347a21acabe80ec0e.jpg", 'Object Detection'],
-                ["examples/liver-test/15_239_203_35_22_jpg.rf.408ffd5311f8d90f5db8f5913e876dc9.jpg", 'Object Detection'],
-                ["examples/liver-test/15_242_202_12_107_jpg.rf.4d864b8288a54e618e6fa0c327e1445e.jpg", 'Object Detection']
+                ["examples/bccd-test/BloodImage_00113_jpg.rf.ab69dfaa52c1b3249cf44fa66afbb619.jpg", 'Object Detection'],
+                ["examples/bccd-test/BloodImage_00120_jpg.rf.4a2f84ca3564ef453b12ceb9c852e32e.jpg", 'Object Detection'],
             ],
             
             inputs=[input_img, task_prompt],