Spaces:
Runtime error
Runtime error
File size: 2,256 Bytes
8a581f5 a79cc2a 8a581f5 a79cc2a 8a581f5 b985e31 f4cffb9 b985e31 212387e d35d17d 8a581f5 597d161 047f126 597d161 047f126 d35d17d 300feb8 496cf33 300feb8 f924a75 8a581f5 d35d17d ca06c20 8a581f5 496cf33 8a581f5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import os
import subprocess
import gradio as gr
# execute a CLI command
def execute_command(command: str) -> None:
subprocess.run(command, check=True)
def infer(video_frames, masks_frames):
video_frames_folder = "inputs/object_removal/bmx-trees"
masks_folder = "inputs/object_removal/bmx-trees_mask"
# Create the "results" folder if it doesn't exist
output_folder = "results"
if not os.path.exists(output_folder):
os.makedirs(output_folder)
#bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
command = [
f"python",
f"inference_propainter.py",
f"--video={video_frames_folder}",
f"--mask={masks_folder}",
f"--output={output_folder}"
]
execute_command(command)
# Get the list of files in the "results" folder
result_files = os.listdir(output_folder)
# Print the content of the "results" folder
print(f"Contents of the {output_folder} folder:")
for item in result_files:
print(item)
# List the content of the "bmx-trees" folder within "results"
bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
bmx_trees_content = [os.path.join(bmx_trees_folder, item) for item in os.listdir(bmx_trees_folder)]
print(f"Contents of the {bmx_trees_folder} folder:")
for item in bmx_trees_content:
print(item)
return "done", bmx_trees_content[1]
css="""
#col-container{
margin: 0 auto;
max-width: 840px;
text-align: left;
}
"""
with gr.Blocks(css=css) as demo:
with gr.Column(elem_id="col-container"):
gr.HTML("""
<h2 style="text-align: center;">ProPainter</h2>
<p style="text-align: center;">
</p>
""")
with gr.Row():
with gr.Column():
video_frames = gr.Files(label="Video frames")
masks_frames = gr.Files(label="Masks frames")
submit_btn = gr.Button("Submit")
with gr.Column():
result = gr.Textbox(label="Result")
res_files = gr.Video()
submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result, res_files])
demo.queue(max_size=12).launch() |