Initial CADCoder Space with Gradio

app.py

@@ -1,14 +1,17 @@
 import os, sys, subprocess
 
+# Path where CADFusion repo should be cloned
+cadfusion_repo = os.path.join(os.path.dirname(__file__), "CADFusion")
+
 # Clone CADFusion repo if not already cloned
-if not os.path.exists("CADFusion"):
-    subprocess.check_call(["git", "clone", "https://github.com/microsoft/CADFusion.git"])
+if not os.path.exists(cadfusion_repo):
+    subprocess.check_call(["git", "clone", "https://github.com/microsoft/CADFusion.git", cadfusion_repo])
 
 # Add CADFusion repo to Python path
-sys.path.append("CADFusion")
+sys.path.append(cadfusion_repo)
 
 # Now import from repo
-from models import CADFusionModel  # repo folder has `models.py`
+from models import CADFusionModel  # models.py lives inside the cloned repo
 
 import gradio as gr
 import torch