Update app.py

app.py

@@ -9,10 +9,11 @@ def install_model(namemodel,tokenn,namemodelonxx):
 def convert_to_onnx(model,namemodelonxx):
       vocab_size = model.text_encoder.embed_tokens.weight.size(0)
       example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
+      x=f"{namemodelonxx}.onnx"
       torch.onnx.export(
           model,  # The model to be exported
           example_input,  # Example input for the model
-          namemodelonxx,  # The filename for the exported ONNX model
+          x,  # The filename for the exported ONNX model
           opset_version=11,  # Use an appropriate ONNX opset version
           input_names=['input'],  # Name of the input layer
           output_names=['output'],  # Name of the output layer
@@ -21,7 +22,7 @@ def convert_to_onnx(model,namemodelonxx):
               'output': {0: 'batch_size'}
           }
       )
-      return namemodelonxx
+      return x
 with gr.Blocks() as demo:
      with gr.Row():
           with gr.Column():