Spaces:
Sleeping
Sleeping
cosmetic changes
Browse files
app.py
CHANGED
|
@@ -103,7 +103,7 @@ def infer(video_frames, masks_frames, project_name):
|
|
| 103 |
for item in results_folder_content:
|
| 104 |
print(item)
|
| 105 |
|
| 106 |
-
return
|
| 107 |
|
| 108 |
|
| 109 |
def get_frames(video_in, img_type):
|
|
@@ -295,7 +295,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
|
|
| 295 |
for item in results_folder_content:
|
| 296 |
print(item)
|
| 297 |
|
| 298 |
-
return
|
| 299 |
|
| 300 |
css="""
|
| 301 |
#col-container{
|
|
@@ -318,7 +318,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 318 |
with gr.Row():
|
| 319 |
with gr.Tab("Manual"):
|
| 320 |
with gr.Column():
|
| 321 |
-
project_name = gr.Textbox(label="Name your project", value="my-new-project")
|
| 322 |
video_frames = gr.File(label="Video frames", file_types=["image"], file_count="multiple")
|
| 323 |
masks_frames = gr.File(label="Masks frames", file_types=["image"], file_count="multiple")
|
| 324 |
|
|
@@ -326,20 +326,20 @@ with gr.Blocks(css=css) as demo:
|
|
| 326 |
|
| 327 |
with gr.Tab("Auto"):
|
| 328 |
with gr.Column():
|
| 329 |
-
project_name_2 = gr.Textbox(label="Name your project", value="my-new-project")
|
| 330 |
video_in = gr.Video(label="Source video", source="upload", format="mp4")
|
| 331 |
subject_to_remove = gr.Textbox(label="Subject to remove")
|
| 332 |
|
| 333 |
submit_auto_btn = gr.Button("Submit")
|
| 334 |
|
| 335 |
with gr.Column():
|
| 336 |
-
|
| 337 |
res_masked = gr.Video(label="Masked video")
|
| 338 |
res_files = gr.Video(label="Final result")
|
| 339 |
|
| 340 |
|
| 341 |
|
| 342 |
-
submit_btn.click(fn=infer, inputs=[video_frames, masks_frames, project_name], outputs=[
|
| 343 |
-
submit_auto_btn.click(fn=infer_auto, inputs=[project_name_2, video_in, subject_to_remove], outputs=[
|
| 344 |
|
| 345 |
demo.queue(max_size=12).launch()
|
|
|
|
| 103 |
for item in results_folder_content:
|
| 104 |
print(item)
|
| 105 |
|
| 106 |
+
return results_folder_content[0], results_folder_content[1]
|
| 107 |
|
| 108 |
|
| 109 |
def get_frames(video_in, img_type):
|
|
|
|
| 295 |
for item in results_folder_content:
|
| 296 |
print(item)
|
| 297 |
|
| 298 |
+
return results_folder_content[0], results_folder_content[1]
|
| 299 |
|
| 300 |
css="""
|
| 301 |
#col-container{
|
|
|
|
| 318 |
with gr.Row():
|
| 319 |
with gr.Tab("Manual"):
|
| 320 |
with gr.Column():
|
| 321 |
+
project_name = gr.Textbox(label="Name your project", info="no spaces nor special characters", value="my-new-project")
|
| 322 |
video_frames = gr.File(label="Video frames", file_types=["image"], file_count="multiple")
|
| 323 |
masks_frames = gr.File(label="Masks frames", file_types=["image"], file_count="multiple")
|
| 324 |
|
|
|
|
| 326 |
|
| 327 |
with gr.Tab("Auto"):
|
| 328 |
with gr.Column():
|
| 329 |
+
project_name_2 = gr.Textbox(label="Name your project", info="no spaces nor special characters", value="my-new-project")
|
| 330 |
video_in = gr.Video(label="Source video", source="upload", format="mp4")
|
| 331 |
subject_to_remove = gr.Textbox(label="Subject to remove")
|
| 332 |
|
| 333 |
submit_auto_btn = gr.Button("Submit")
|
| 334 |
|
| 335 |
with gr.Column():
|
| 336 |
+
gr.Markdown("""###Results""")
|
| 337 |
res_masked = gr.Video(label="Masked video")
|
| 338 |
res_files = gr.Video(label="Final result")
|
| 339 |
|
| 340 |
|
| 341 |
|
| 342 |
+
submit_btn.click(fn=infer, inputs=[video_frames, masks_frames, project_name], outputs=[res_masked, res_files])
|
| 343 |
+
submit_auto_btn.click(fn=infer_auto, inputs=[project_name_2, video_in, subject_to_remove], outputs=[res_masked, res_files])
|
| 344 |
|
| 345 |
demo.queue(max_size=12).launch()
|