Trying to use ONNX model.

app.py

@@ -5,7 +5,7 @@ from transformers import AutoTokenizer
 from transformers import pipeline
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from optimum.onnxruntime import ORTModelForQuestionAnswering
+from optimum.onnxruntime import ORTModelForSequenceClassification
 
 # CORS Config
 app = FastAPI()
@@ -19,7 +19,7 @@ app.add_middleware(
 )
 
 model_name = "xenova/mobilebert-uncased-mnli"
-model = ORTModelForQuestionAnswering.from_pretrained(model_name, file_name="onnx/model.onnx")
+model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name="onnx/model.onnx")
 tokenizer = AutoTokenizer.from_pretrained("typeform/mobilebert-uncased-mnli", model_max_length=512)
 
 # file = cached_download("https://huggingface.co/" + model_name + "")