Download LDraw library instead of keeping it in the repo

.gitignore

@@ -9,5 +9,13 @@ wheels/
 # Virtual environments
 .venv
 
+# Gurobi files
+gurobi.lic
+
+# LDraw files
+ldraw
+complete.zip
+
 .idea
-.gradio
+.gradio
+out

app.py

@@ -1,29 +1,34 @@
 import os
 import subprocess
+import zipfile
 from dataclasses import fields
+from urllib.request import urlretrieve
 
 import gradio as gr
 import transformers
 from legogpt.models import LegoGPT, LegoGPTConfig
-import zipfile
 
 
 def setup():
     # Set up Gurobi licence
+    licence_filename = 'gurobi.lic'
     licence_lines = []
     for secret_name in ['WLSACCESSID', 'WLSSECRET', 'LICENSEID']:
         secret = os.environ.get(secret_name)
         if not secret:
             raise ValueError(f'Env variable {secret_name} not found. Please set it in the Hugging Face Space settings.')
         licence_lines.append(f'{secret_name}={secret}\n')
-    with open('gurobi.lic', 'w') as f:
+    with open(licence_filename, 'w') as f:
         f.writelines(licence_lines)
-    os.environ['GRB_LICENSE_FILE'] = os.path.abspath('gurobi.lic')
+    os.environ['GRB_LICENSE_FILE'] = os.path.abspath(licence_filename)
 
-    # Set LDraw library path
-    with zipfile.ZipFile('ldraw.zip') as zip_ref:
-        zip_ref.extractall('ImportLDraw')
-    os.environ['LDRAW_LIBRARY_PATH'] = os.path.abspath('ImportLDraw/ldraw')
+    # Download LDraw part library and set LDraw library path
+    ldraw_zip_url = 'https://library.ldraw.org/library/updates/complete.zip'
+    ldraw_zip_filename = 'complete.zip'
+    urlretrieve(ldraw_zip_url, ldraw_zip_filename)
+    with zipfile.ZipFile(ldraw_zip_filename) as zip_ref:
+        zip_ref.extractall()
+    os.environ['LDRAW_LIBRARY_PATH'] = os.path.abspath('ldraw')
 
 
 def main():
@@ -57,8 +62,11 @@ def main():
 
         # Render results and write to files
         render_lego_filename = 'render_lego.py'
-        ldr_filename = os.path.abspath('output.ldr')
-        img_filename = os.path.abspath('output.png')
+
+        output_dir = os.path.abspath('out')
+        os.makedirs(output_dir, exist_ok=True)
+        ldr_filename = os.path.join(output_dir, 'output.ldr')
+        img_filename = os.path.join(output_dir, 'output.png')
         with open(ldr_filename, 'w') as f:
             f.write(output['lego'].to_ldr())
 

ldraw.zip

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3115747ca75879100a799ebabd9bf25302b170828bda39ab3dd9379a26ae3d07
-size 99633074