Update app.py

app.py

@@ -65,11 +65,11 @@ class OnnxModelConverter:
         """
         model = VitsModel.from_pretrained(model_name, token=token)
         onnx_file = f"/tmp/{onnx_filename}.onnx"
-        vocab_size = model.text_encoder.embed_tokens.weight.size(0)
-        example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
+        example_input  = torch.randn(1, 192, 10)
+
 
         torch.onnx.export(
-            model,
+            model.decoder,
             example_input,
             onnx_file,
             opset_version=11,