Paolo-Fraccaro
commited on
Commit
•
009060b
1
Parent(s):
875b225
try to fix path issue
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
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)
|