Hugging Face's logo

Spaces:
justinj92
/
florence-2
Sleeping

App Files Community
florence-2
File size: 2,552 Bytes 
39bd209
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
 
OBJECT_DETECTION_TASK_NAME = "Object Detection"
REGION_PROPOSAL_TASK_NAME = "Region Proposal"
DENSE_REGION_CAPTION_TASK_NAME = "Dense Region Caption"
CAPTION_TASK_NAME = "Caption"
DETAILED_CAPTION_TASK_NAME = "Detailed Caption"
MORE_DETAILED_CAPTION_TASK_NAME = "More Detailed Caption"
OCR_TASK_NAME = "OCR"
OCR_WITH_REGION_TASK_NAME = "OCR with Region"
REGION_TO_CATEGORY_TASK_NAME = "Region to Category"
REGION_TO_DESCRIPTION_TASK_NAME = "Region to Description"

TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME,
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]
TASKS = {
    OBJECT_DETECTION_TASK_NAME: "<OD>",
    REGION_PROPOSAL_TASK_NAME: "<REGION_PROPOSAL>",
    DENSE_REGION_CAPTION_TASK_NAME: "<DENSE_REGION_CAPTION>",
    CAPTION_TASK_NAME: "<CAPTION>",
    DETAILED_CAPTION_TASK_NAME: "<DETAILED_CAPTION>",
    MORE_DETAILED_CAPTION_TASK_NAME: "<MORE_DETAILED_CAPTION>",
    OCR_TASK_NAME: "<OCR>",
    OCR_WITH_REGION_TASK_NAME: "<OCR_WITH_REGION>",
    REGION_TO_CATEGORY_TASK_NAME: "<REGION_TO_CATEGORY>",
    REGION_TO_DESCRIPTION_TASK_NAME: "<REGION_TO_DESCRIPTION>"
}
IMAGE_INPUT_TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME,
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
]
IMAGE_PROMPTER_INPUT_TASK_NAMES = [
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]
IMAGE_OUTPUT_TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]
TEXTBOX_OUTPUT_TASK_NAMES = [
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME
]
IMAGE_TO_IMAGE_TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME
]
IMAGE_TO_TEXT_TASK_NAMES = [
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME
]
IMAGE_PROMPT_TO_IMAGE_TASK_NAMES = [
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]