recorrect onnx to device

mmxai/onnx/onnxModel.py

@@ -11,7 +11,7 @@ from PIL import Image
 from transformers import BertTokenizer, AutoTokenizer, BertTokenizerFast, XLNetTokenizer
 from mmf.models.mmbt import MMBT
 
-class ONNXInterface(torch.nn.Module):
+class ONNXInterface:
 	def __init__(self,model_path,tokenizer = None):
 		self.model = onnx.load(model_path)
 		self.ort_session = onnxruntime.InferenceSession(model_path)
@@ -92,7 +92,9 @@ class ONNXInterface(torch.nn.Module):
 
 		ort_outs = self.ort_session.run([output_name], ort_inputs)
 		return ort_outs
-
+	def to(self,decive):
+		if self.defaultmodel != None:
+			self.defaultmodel.to(device)
 	def classify(self,image,text_input, image_tensor = None):
 		'''
 		Args: