Hugging Face's logo

Spaces:
supersolar
/
florence-sam-kaggle
Runtime error

App Files Community
supersolar commited on
Commit
78cb3e1
·
verified ·
1 Parent(s): 901914b

Update utils/florence.py

Browse files
Files changed (1) hide show
  1. utils/florence.py +2 -2
utils/florence.py CHANGED
@@ -7,8 +7,8 @@ from PIL import Image
7
  from transformers import AutoModelForCausalLM, AutoProcessor
8
  from transformers.dynamic_module_utils import get_imports
9
 
10
- FLORENCE_CHECKPOINT = "microsoft/Florence-2-base"
11
- FLORENCE_CHECKPOINT = "microsoft/Florence-2-large-ft"
12
  FLORENCE_OBJECT_DETECTION_TASK = '<OD>'
13
  FLORENCE_DETAILED_CAPTION_TASK = '<MORE_DETAILED_CAPTION>'
14
  FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK = '<CAPTION_TO_PHRASE_GROUNDING>'
 
7
  from transformers import AutoModelForCausalLM, AutoProcessor
8
  from transformers.dynamic_module_utils import get_imports
9
 
10
+ FLORENCE_CHECKPOINT = "microsoft/Florence-2-large"
11
+ #FLORENCE_CHECKPOINT = "microsoft/Florence-2-large-ft"
12
  FLORENCE_OBJECT_DETECTION_TASK = '<OD>'
13
  FLORENCE_DETAILED_CAPTION_TASK = '<MORE_DETAILED_CAPTION>'
14
  FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK = '<CAPTION_TO_PHRASE_GROUNDING>'