diff options
Diffstat (limited to 'app/dock-single.py')
| -rw-r--r-- | app/dock-single.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/app/dock-single.py b/app/dock-single.py index c3f0cb5..a94f3e9 100644 --- a/app/dock-single.py +++ b/app/dock-single.py @@ -244,7 +244,11 @@ with tempfile.TemporaryDirectory() as directory: get3DModel(pdbpath,"%s_out.pdbqt"%(records[4])) os.system("collada2gltf -i model.dae -o model.gltf") copyfile("model.gltf",os.path.join(modelDirectory,(str(jobID)+".gltf"))) - os.system("docker run -it --rm -v $(pwd):/usr/app leon/usd-from-gltf:latest model.gltf model.usdz") + arch = os.popen("uname -m").read() + if "x86" in arch: + os.system("docker run -it --rm -v $(pwd):/usr/app leon/usd-from-gltf:latest model.gltf model.usdz") + elif "aarch64" in arch: + os.system("docker run -it --rm -v $(pwd):/usr/app navanchauhan/usd-from-gltf:latest model.gltf model.usdz") try: copyfile("model.usdz",os.path.join(modelDirectory,(str(jobID)+".usdz"))) except: |