Spaces:
Running
on
L40S
Running
on
L40S
Update app.py
Browse files
app.py
CHANGED
@@ -81,8 +81,8 @@ def run_inference(temp_dir, removed_bg_path):
|
|
81 |
)
|
82 |
|
83 |
|
84 |
-
# Retrieve the file name
|
85 |
-
removed_bg_file_name = os.path.basename(removed_bg_path)
|
86 |
output_videos = glob(os.path.join(f"out/{removed_bg_file_name}", "*.mp4"))
|
87 |
return output_videos
|
88 |
except subprocess.CalledProcessError as e:
|
|
|
81 |
)
|
82 |
|
83 |
|
84 |
+
# Retrieve the file name without the extension
|
85 |
+
removed_bg_file_name = os.path.splitext(os.path.basename(removed_bg_path))[0]
|
86 |
output_videos = glob(os.path.join(f"out/{removed_bg_file_name}", "*.mp4"))
|
87 |
return output_videos
|
88 |
except subprocess.CalledProcessError as e:
|