huzey commited on
Commit
da6c997
1 Parent(s): f1aaf40
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -491,7 +491,7 @@ def run_fn(
491
  # resize the images before acquiring GPU
492
  if "AlignedThreeModelAttnNodes" == model_name:
493
  # dirty patch for the alignedcut paper
494
- resolution = (672, 672)
495
  else:
496
  resolution = RES_DICT[model_name]
497
  images = [tup[0] for tup in images]
@@ -1021,7 +1021,7 @@ with demo:
1021
  gr.Markdown('**Features are aligned across models and layers.** A linear alignment transform is trained for each model/layer, learning signal comes from 1) fMRI brain activation and 2) segmentation preserving eigen-constraints.')
1022
  gr.Markdown('NCUT is computed on the concatenated graph of all models, layers, and images. Color is **aligned** across all models and layers.')
1023
  gr.Markdown('')
1024
- gr.Markdown("To see a good pattern, you will need to load 100 images. Running out of HuggingFace GPU Quota? Try [Demo](https://ncut-pytorch.readthedocs.io/en/latest/demo/) hosted at UPenn")
1025
  gr.Markdown('---')
1026
  with gr.Row():
1027
  with gr.Column(scale=5, min_width=200):
 
491
  # resize the images before acquiring GPU
492
  if "AlignedThreeModelAttnNodes" == model_name:
493
  # dirty patch for the alignedcut paper
494
+ resolution = (224, 224)
495
  else:
496
  resolution = RES_DICT[model_name]
497
  images = [tup[0] for tup in images]
 
1021
  gr.Markdown('**Features are aligned across models and layers.** A linear alignment transform is trained for each model/layer, learning signal comes from 1) fMRI brain activation and 2) segmentation preserving eigen-constraints.')
1022
  gr.Markdown('NCUT is computed on the concatenated graph of all models, layers, and images. Color is **aligned** across all models and layers.')
1023
  gr.Markdown('')
1024
+ gr.Markdown("To see a good pattern, you will need to load 100~1000 images. 100 images need 10sec for RTX4090. Running out of HuggingFace GPU Quota? Try [Demo](https://ncut-pytorch.readthedocs.io/en/latest/demo/) hosted at UPenn")
1025
  gr.Markdown('---')
1026
  with gr.Row():
1027
  with gr.Column(scale=5, min_width=200):