katuni4ka commited on
Commit
0a10233
·
verified ·
1 Parent(s): a20c544

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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