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 |
|