Update app.py

app.py

@@ -104,15 +104,13 @@ class OnnxModelConverter:
         # Trace the decoder part of the model
         example_inputs = inputs.input_ids.type(torch.LongTensor) 
 
-        torch.onnx.export(
-            model,
-            example_inputs,
-            onnx_file,
-            opset_version=11,
-            input_names=['input'],
-            output_names=['output'],
-            dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
-        )
+        torch.onnx.export(model,
+                  example_inputs,
+                  onnx_file,
+                  input_names=["input"],
+                  output_names=["output"],
+                  dynamic_axes={"input": {0: "batch_size", 1: "sequence_length"},
+                                "output": {0: "batch_size", 1: "sequence_length"}})
 
         return  self.download_file(onnx_file)