Update for DeTikZify v2

Dockerfile

@@ -68,13 +68,17 @@ RUN pyenv install 3.11 \
   && pyenv global 3.11 \
   && pyenv rehash \
   && pip install --no-cache-dir --upgrade setuptools wheel \
-  && pip install --no-cache-dir 'detikzify @ git+https://github.com/potamides/DeTikZify' \
+  && pip install --no-cache-dir 'detikzify[legacy] @ git+https://github.com/potamides/DeTikZify' \
   && pip install --no-cache-dir flash-attn --no-build-isolation \
   && pip install --no-cache-dir spaces
 
 RUN chown -R user $HOME/app
 COPY --chown=user . $HOME/app
 
+## preload v2 models
+# -----------------------------------------------------------------------------
+RUN huggingface-cli download nllg/detikzify-v2-8b
+
 ## launch app
 # -----------------------------------------------------------------------------
 USER user

README.md

@@ -7,11 +7,6 @@ sdk: docker
 pinned: false
 license: apache-2.0
 suggested_hardware: a10g-small
-startup_duration_timeout: 1h
-preload_from_hub:
-  - timm/ViT-SO400M-14-SigLIP-384
-  - nllg/detikzify-ds-7b
-  - nllg/detikzify-ds-1.3b
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -32,7 +32,7 @@ if is_official_demo() and not cuda.is_available():
         gr.HTML(f'{BANNER}\n<p>{message}</p>\n{html}')
 else:
     use_big_models = cuda.is_available() and cuda.get_device_properties(0).total_memory > 15835660288
-    model = f"detikzify-ds-{'7' if use_big_models else '1.3'}b"
+    model = f"detikzify-{'v2-8b' if use_big_models else 'ds-1.3b'}"
     demo = build_ui(lock=is_official_demo(), model=model, light=True).queue()
     # Hack to temporarily work around memory leak, see:
     #   * https://huggingface.co/spaces/nllg/DeTikZify/discussions/2