Pouriarouzrokh commited on
Commit
48fd666
·
1 Parent(s): 0f98ba7

added cache examples

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -49,7 +49,7 @@ def create_ds(img_paths):
49
  ]
50
  return mn.data.Dataset(data_list, transform=mn.transforms.Compose(Ts_list))
51
 
52
- def make_predictions(img_path, angles=None, cls_batch=None, rotate_to_standard=False, sampler="PLMS5"):
53
 
54
  global model
55
  global BASELINE_NOISE
@@ -131,13 +131,17 @@ with gr.Blocks(css=css_style) as app:
131
  examples = [os.path.join("./data/examples", f) for f in os.listdir("./data/examples") if "xr" in f],
132
  inputs = [input_img],
133
  label = "Xray Examples",
134
- elem_id='examples'
 
 
135
  )
136
  gr.Examples(
137
  examples = [os.path.join("./data/examples", f) for f in os.listdir("./data/examples") if "drr" in f],
138
  inputs = [input_img],
139
  label = "DRR Examples",
140
- elem_id='examples'
 
 
141
  )
142
  with gr.Row():
143
  gr.Markdown('Please select an example image, choose your rotation angles, and press Rotate!', elem_classes='text')
 
49
  ]
50
  return mn.data.Dataset(data_list, transform=mn.transforms.Compose(Ts_list))
51
 
52
+ def make_predictions(img_path, angles=None, cls_batch=None, rotate_to_standard=False, sampler="DDIM5"):
53
 
54
  global model
55
  global BASELINE_NOISE
 
131
  examples = [os.path.join("./data/examples", f) for f in os.listdir("./data/examples") if "xr" in f],
132
  inputs = [input_img],
133
  label = "Xray Examples",
134
+ elem_id='examples',
135
+ cache_examples=True,
136
+ outputs=output_img
137
  )
138
  gr.Examples(
139
  examples = [os.path.join("./data/examples", f) for f in os.listdir("./data/examples") if "drr" in f],
140
  inputs = [input_img],
141
  label = "DRR Examples",
142
+ elem_id='examples',
143
+ cache_examples=True,
144
+ outputs=output_img
145
  )
146
  with gr.Row():
147
  gr.Markdown('Please select an example image, choose your rotation angles, and press Rotate!', elem_classes='text')