fffiloni commited on
Commit
e1a1acd
1 Parent(s): c5993c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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: