Commit
·
cf690ac
1
Parent(s):
51e5f45
up
Browse files
app.py
CHANGED
@@ -20,9 +20,9 @@ from transformers.models.auto.modeling_auto import (
|
|
20 |
MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING_NAMES,
|
21 |
)
|
22 |
|
23 |
-
audio_models = MODEL_FOR_CTC_MAPPING_NAMES.keys() + MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES.keys() + MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES.keys()
|
24 |
|
25 |
-
vision_models = MODEL_FOR_VISION_2_SEQ_MAPPING_NAMES.keys() + MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING_NAMES.keys() + MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING_NAMES.keys() + MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES.keys() + MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING_NAMES.keys() + MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING_NAMES.keys()
|
26 |
|
27 |
today = datetime.date.today()
|
28 |
year, week, _ = today.isocalendar()
|
|
|
20 |
MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING_NAMES,
|
21 |
)
|
22 |
|
23 |
+
audio_models = list(MODEL_FOR_CTC_MAPPING_NAMES.keys()) + list(MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES.keys()) + list(MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES.keys())
|
24 |
|
25 |
+
vision_models = list(MODEL_FOR_VISION_2_SEQ_MAPPING_NAMES.keys()) + list(MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING_NAMES.keys()) + list(MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING_NAMES.keys()) + list(MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES.keys()) + list(MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING_NAMES.keys()) + list(MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING_NAMES.keys())
|
26 |
|
27 |
today = datetime.date.today()
|
28 |
year, week, _ = today.isocalendar()
|