update web
Browse files
app.py
CHANGED
|
@@ -76,8 +76,6 @@ def load_model_if_needed(variant: str):
|
|
| 76 |
log.info(f"Model {variant} loaded and cached successfully")
|
| 77 |
return net, feature_utils
|
| 78 |
|
| 79 |
-
# ensure_models_downloaded()
|
| 80 |
-
|
| 81 |
|
| 82 |
@spaces.GPU(duration=60)
|
| 83 |
@torch.inference_mode()
|
|
@@ -254,6 +252,7 @@ gr_interface = gr.Interface(
|
|
| 254 |
cache_examples="lazy", # Turn on to cache.
|
| 255 |
)
|
| 256 |
|
|
|
|
| 257 |
gr_interface.queue(15).launch()
|
| 258 |
|
| 259 |
# theme = gr.themes.Soft(
|
|
|
|
| 76 |
log.info(f"Model {variant} loaded and cached successfully")
|
| 77 |
return net, feature_utils
|
| 78 |
|
|
|
|
|
|
|
| 79 |
|
| 80 |
@spaces.GPU(duration=60)
|
| 81 |
@torch.inference_mode()
|
|
|
|
| 252 |
cache_examples="lazy", # Turn on to cache.
|
| 253 |
)
|
| 254 |
|
| 255 |
+
ensure_models_downloaded()
|
| 256 |
gr_interface.queue(15).launch()
|
| 257 |
|
| 258 |
# theme = gr.themes.Soft(
|