Paolo-Fraccaro commited on
Commit
009060b
1 Parent(s): 875b225

try to fix path issue

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -247,7 +247,12 @@ def extract_rgb_imgs(input_img, rec_img, mask_img, channels, mean, std):
247
 
248
  def predict_on_images(data_files: list, mask_ratio: float, yaml_file_path: str, checkpoint: str):
249
 
250
- # os.makedirs(output_dir, exist_ok=True)
 
 
 
 
 
251
 
252
  # Get parameters --------
253
  print('This is the printout', data_files)
 
247
 
248
  def predict_on_images(data_files: list, mask_ratio: float, yaml_file_path: str, checkpoint: str):
249
 
250
+
251
+ try:
252
+ data_files = [x.name for x in data_files]
253
+ print('Path extracted from example')
254
+ except:
255
+ print('Files submitted through UI')
256
 
257
  # Get parameters --------
258
  print('This is the printout', data_files)