Update app.py

app.py

@@ -18,22 +18,20 @@ 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(),
-    #'microsoft/Florence-2-large': AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-large', trust_remote_code=True).to("cuda").eval(), # GPU memory use is too high to run on Kaggle
+    'mroca/Florence-2-base-DocVQA-ft': AutoModelForCausalLM.from_pretrained('mroca/Florence-2-base-DocVQA-ft', trust_remote_code=True).to("cuda").eval(),
     'microsoft/Florence-2-base-ft': AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base-ft', trust_remote_code=True).to("cuda").eval(),
     'microsoft/Florence-2-base': AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to("cuda").eval(),
-    'mroca/Florence-2-base-DocVQA-ft': AutoModelForCausalLM.from_pretrained('mroca/Florence-2-base-DocVQA-ft', trust_remote_code=True).to("cuda").eval(), # Finetuned using https://github.com/mroca1/florence2-finetuning/tree/main
 }
 
 processors = {
     'microsoft/Florence-2-large-ft': AutoProcessor.from_pretrained('microsoft/Florence-2-large-ft', trust_remote_code=True),
-    #'microsoft/Florence-2-large': AutoProcessor.from_pretrained('microsoft/Florence-2-large', trust_remote_code=True),
+    'mroca/Florence-2-base-DocVQA-ft': AutoProcessor.from_pretrained('mroca/Florence-2-base-DocVQA-ft', trust_remote_code=True),
     'microsoft/Florence-2-base-ft': AutoProcessor.from_pretrained('microsoft/Florence-2-base-ft', trust_remote_code=True),
     'microsoft/Florence-2-base': AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True),
-    'mroca/Florence-2-base-DocVQA-ft': AutoModelForCausalLM.from_pretrained('mroca/Florence-2-base-DocVQA-ft', trust_remote_code=True).to("cuda").eval(),
 }
 
 
-DESCRIPTION = "# [Florence-2 Demo](https://huggingface.co/mroca/Florence-2-base-DocVQA-ft)"
+DESCRIPTION = "# [Florence-2 Demo](https://huggingface.co/microsoft/Florence-2-large)"
 
 colormap = ['blue','orange','green','purple','brown','pink','gray','olive','cyan','red',
             'lime','indigo','violet','aqua','magenta','coral','gold','tan','skyblue']
@@ -283,4 +281,4 @@ with gr.Blocks(css=css) as demo:
 
         submit_btn.click(process_image, [input_img, task_prompt, text_input, model_selector], [output_text, output_img])
 
-demo.launch(share=False)
+demo.launch(share=False)