potamides commited on
Commit
2e7a270
·
1 Parent(s): 31b00a9

Update for DeTikZify v2.5

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. app.py +1 -3
Dockerfile CHANGED
@@ -77,7 +77,7 @@ COPY --chown=user . $HOME/app
77
 
78
  ## preload v2 models
79
  # -----------------------------------------------------------------------------
80
- RUN huggingface-cli download nllg/detikzify-v2-8b \
81
  && chown -R user $HOME/.cache
82
 
83
  ## launch app
 
77
 
78
  ## preload v2 models
79
  # -----------------------------------------------------------------------------
80
+ RUN huggingface-cli download nllg/detikzify-v2.5-8b \
81
  && chown -R user $HOME/.cache
82
 
83
  ## launch app
app.py CHANGED
@@ -31,9 +31,7 @@ if is_official_demo() and not cuda.is_available():
31
  """)
32
  gr.HTML(f'{BANNER}\n<p>{message}</p>\n{html}')
33
  else:
34
- use_big_models = cuda.is_available() and cuda.get_device_properties(0).total_memory > 15835660288
35
- model = f"detikzify-{'v2-8b' if use_big_models else 'ds-1.3b'}"
36
- demo = build_ui(lock=is_official_demo(), model=model, light=True).queue()
37
  # Hack to temporarily work around memory leak, see:
38
  # * https://huggingface.co/spaces/nllg/DeTikZify/discussions/2
39
  # * https://github.com/gradio-app/gradio/issues/8503
 
31
  """)
32
  gr.HTML(f'{BANNER}\n<p>{message}</p>\n{html}')
33
  else:
34
+ demo = build_ui(lock=is_official_demo(), light=True).queue()
 
 
35
  # Hack to temporarily work around memory leak, see:
36
  # * https://huggingface.co/spaces/nllg/DeTikZify/discussions/2
37
  # * https://github.com/gradio-app/gradio/issues/8503