Update utils/florencegpu2.py

utils/florencegpu2.py

@@ -7,7 +7,8 @@ from PIL import Image
 from transformers import AutoModelForCausalLM, AutoProcessor
 from transformers.dynamic_module_utils import get_imports
 
-FLORENCE_CHECKPOINT = "microsoft/Florence-2-base"
+FLORENCE_CHECKPOINT = "microsoft/Florence-2-large"
+#FLORENCE_CHECKPOINT = "microsoft/Florence-2-large-ft"
 FLORENCE_OBJECT_DETECTION_TASK = '<OD>'
 FLORENCE_DETAILED_CAPTION_TASK = '<MORE_DETAILED_CAPTION>'
 FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK = '<CAPTION_TO_PHRASE_GROUNDING>'
@@ -54,4 +55,4 @@ def run_florence_inference(
         generated_ids, skip_special_tokens=False)[0]
     response = processor.post_process_generation(
         generated_text, task=task, image_size=image.size)
-    return generated_text, response
+    return generated_text, response