Update app.py

app.py

@@ -9,8 +9,11 @@ class OnnxModelConverter:
     def __init__(self):
         self.model = None
     def download_file(self,file_path):
-         with open("file.name", "wb") as f:
-             return  file.read()
+         if not os.path.exists(file_path):
+         with open(file_path, "w") as file:
+              file.write("This is a test ONNX model.")
+         return gr.File(file_path, label="Download ONNX File")
+          
 
     def convert(self, model_name, token, onnx_filename, conversion_type):
         """
@@ -61,7 +64,7 @@ class OnnxModelConverter:
             dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
         )
 
-        return  onnx_file
+        return  self.download_file(onnx_file)
 
 
     def convert_only_decoder(self, model_name, token, onnx_filename):
@@ -92,7 +95,7 @@ class OnnxModelConverter:
             dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
         )
 
-        return  onnx_file
+        return  self.download_file(onnx_file)
 
     def convert_full_model(self, model_name, token, onnx_filename):
         """
@@ -122,7 +125,7 @@ class OnnxModelConverter:
             dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
         )
 
-        return  onnx_file
+        return  self.download_file(onnx_file)
     def starrt(self):
         with gr.Blocks() as demo:
             with gr.Row():