Spaces:
Paused
Paused
Update app.py
Browse files
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 |
-
|
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 |
-
|
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 = "
|
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 |
|