Update app.py

app.py

@@ -15,21 +15,6 @@ import numpy as np
 
 import gc
 
-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(),
-    #'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(),
-}
-
-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),
-    #'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),
-}
-
-
 DESCRIPTION = "# [Florence-2 Demo](https://huggingface.co/microsoft/Florence-2-large)"
 
 colormap = ['blue','orange','green','purple','brown','pink','gray','olive','cyan','red',
@@ -42,9 +27,8 @@ def fig_to_pil(fig):
     return Image.open(buf)
 
 model_id='microsoft/Florence-2-large'
-model = models[model_id]
-processor = processors[model_id]
-processor
+model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-large', trust_remote_code=True).to("cuda").eval()
+processor = AutoProcessor.from_pretrained('microsoft/Florence-2-large', trust_remote_code=True)
 
 @spaces.GPU
 def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florence-2-large'):