return object data

app.py

@@ -166,9 +166,9 @@ from pathlib import Path
 # Function to visualize the model
 def visualize_model(model_file):
     # Save the uploaded file
-    file_path = os.path.join(os.getcwd(), "weight_files/yolov5.onnx") 
+    file_path = f"./models/{model_file.name}"
     with open(file_path, "wb") as f:
-        f.write(model_file.read())
+        f.write(model_file.data)  # FIXED: Use 'data' instead of 'read()'
 
     # Start Netron server to serve the file
     port = 8081
@@ -195,5 +195,5 @@ demo = gr.Interface(
     description="Upload a model file to visualize its architecture using Netron.",
 )
 
-# Launch the app
-demo.launch(server_name="0.0.0.0", server_port=7860)
+# Launch the app with public link support
+demo.launch(server_name="0.0.0.0", server_port=7860, share=True)