Update app.py

app.py

@@ -5,6 +5,9 @@ from VitsModelSplit.vits_model_only_d import  Vits_models_only_decoder
 from VitsModelSplit.vits_model import VitsModel
 import gradio as gr
 import os
+from transformers import AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar")
+
 def create_file(file_path):
     # مسار الملف المؤقت
     #file_path = "DDFGDdd.onnx"
@@ -96,12 +99,14 @@ class OnnxModelConverter:
         model = Vits_models_only_decoder.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)
+        inputs = tokenizer("السلام عليكم كيف الحال", return_tensors="pt")
+        
+        # Trace the decoder part of the model
+        example_inputs = inputs.input_ids.type(torch.LongTensor) 
 
         torch.onnx.export(
             model,
-            example_input,
+            example_inputs,
             onnx_file,
             opset_version=11,
             input_names=['input'],