handle return type

app.py

@@ -158,7 +158,6 @@ with gr.Blocks(css=custom_css) as interface:
 if __name__ == "__main__":
     interface.launch(share=True)
 """
-
 import gradio as gr
 import os
 import subprocess
@@ -166,13 +165,16 @@ import subprocess
 # Function to visualize the model
 def visualize_model(model_file):
     # Create the output directory if it doesn't exist
-    output_dir = os.path.join("./models", os.path.dirname(model_file.name))
-    os.makedirs(output_dir, exist_ok=True)  # FIXED: Create nested directories if needed
+    output_dir = "./models/tmp/gradio/"
+    os.makedirs(output_dir, exist_ok=True)  # Create directories if needed
+
+    # Define the file path to save
+    file_path = os.path.join(output_dir, model_file.name)
 
     # Save the uploaded file
-    file_path = os.path.join("./models", model_file.name)
     with open(file_path, "wb") as f:
-        f.write(model_file.data)  # Save the uploaded file data
+        with open(model_file.name, "rb") as uploaded_file:
+            f.write(uploaded_file.read())  # FIXED: Properly read and write file data
 
     # Start Netron server to visualize the model
     port = 8081