fffiloni commited on
Commit
5e2c5e4
1 Parent(s): cb1c716

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -171,7 +171,7 @@ def get_matte(video_in, subject_to_remove):
171
 
172
 
173
  def infer_auto(project_name, video_in, subject_to_remove):
174
-
175
  print(video_in)
176
  matte_video = get_matte(video_in, subject_to_remove)
177
 
@@ -188,7 +188,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
188
  os.makedirs(my_video_directory)
189
  else:
190
  # If the directory already exists, add a timestamp to the new directory name
191
- timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
192
  my_video_directory = f"{project_name}_{timestamp}"
193
  os.makedirs(my_video_directory)
194
  print(f"Created the dir: {my_video_directory}")
@@ -215,7 +215,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
215
  os.makedirs(my_masks_directory)
216
  else:
217
  # If the directory already exists, add a timestamp to the new directory name
218
- timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
219
  my_masks_directory = f"{project_name}_masks_{timestamp}"
220
  os.makedirs(my_masks_directory)
221
  print(f"Created the dir: {my_masks_directory}")
@@ -241,7 +241,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
241
  masks_folder = f"{my_masks_directory}"
242
 
243
  # Create the "results" folder if it doesn't exist
244
- output_folder = "results"
245
  if not os.path.exists(output_folder):
246
  os.makedirs(output_folder)
247
 
 
171
 
172
 
173
  def infer_auto(project_name, video_in, subject_to_remove):
174
+ timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
175
  print(video_in)
176
  matte_video = get_matte(video_in, subject_to_remove)
177
 
 
188
  os.makedirs(my_video_directory)
189
  else:
190
  # If the directory already exists, add a timestamp to the new directory name
191
+
192
  my_video_directory = f"{project_name}_{timestamp}"
193
  os.makedirs(my_video_directory)
194
  print(f"Created the dir: {my_video_directory}")
 
215
  os.makedirs(my_masks_directory)
216
  else:
217
  # If the directory already exists, add a timestamp to the new directory name
218
+
219
  my_masks_directory = f"{project_name}_masks_{timestamp}"
220
  os.makedirs(my_masks_directory)
221
  print(f"Created the dir: {my_masks_directory}")
 
241
  masks_folder = f"{my_masks_directory}"
242
 
243
  # Create the "results" folder if it doesn't exist
244
+ output_folder = f"results_{timestamp}"
245
  if not os.path.exists(output_folder):
246
  os.makedirs(output_folder)
247