Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,6 +25,7 @@ from optimum.intel import (
|
|
25 |
OVModelForSequenceClassification,
|
26 |
OVModelForTokenClassification,
|
27 |
OVModelForPix2Struct,
|
|
|
28 |
OVWeightQuantizationConfig,
|
29 |
OVDiffusionPipeline,
|
30 |
)
|
@@ -39,6 +40,7 @@ _HEAD_TO_AUTOMODELS = {
|
|
39 |
"question-answering": "OVModelForQuestionAnswering",
|
40 |
"image-classification": "OVModelForImageClassification",
|
41 |
"audio-classification": "OVModelForAudioClassification",
|
|
|
42 |
}
|
43 |
|
44 |
|
|
|
25 |
OVModelForSequenceClassification,
|
26 |
OVModelForTokenClassification,
|
27 |
OVModelForPix2Struct,
|
28 |
+
OVModelForVisualCausalLM,
|
29 |
OVWeightQuantizationConfig,
|
30 |
OVDiffusionPipeline,
|
31 |
)
|
|
|
40 |
"question-answering": "OVModelForQuestionAnswering",
|
41 |
"image-classification": "OVModelForImageClassification",
|
42 |
"audio-classification": "OVModelForAudioClassification",
|
43 |
+
"image-text-to-text": "OVModelForVisualCausalLM"
|
44 |
}
|
45 |
|
46 |
|