wasmdashai commited on
Commit
81c9dad
·
verified ·
1 Parent(s): a0a07a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ class model_onxx:
76
  torch.onnx.export(
77
  model, # The model to be exported
78
  example_input, # Example input for the model
79
- x, # The filename for the exported ONNX model
80
  opset_version=11, # Use an appropriate ONNX opset version
81
  input_names=['input'], # Name of the input layer
82
  output_names=['output'], # Name of the output layer
 
76
  torch.onnx.export(
77
  model, # The model to be exported
78
  example_input, # Example input for the model
79
+ x,# The filename for the exported ONNX model
80
  opset_version=11, # Use an appropriate ONNX opset version
81
  input_names=['input'], # Name of the input layer
82
  output_names=['output'], # Name of the output layer