Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,8 +9,14 @@ pix2pix = gr.Blocks.load(name="spaces/fffiloni/instruct-pix2pix-clone", api_key=
|
|
| 9 |
|
| 10 |
def get_frames(video_in):
|
| 11 |
frames = []
|
| 12 |
-
#
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
fps = cap.get(cv2.CAP_PROP_FPS)
|
| 15 |
i=0
|
| 16 |
while(cap.isOpened()):
|
|
@@ -53,12 +59,43 @@ def infer(prompt,video_in, seed_in, trim_value):
|
|
| 53 |
|
| 54 |
return final_vid
|
| 55 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
with gr.Blocks(css='style.css') as demo:
|
| 58 |
with gr.Column(elem_id="col-container"):
|
|
|
|
| 59 |
with gr.Row():
|
| 60 |
with gr.Column():
|
| 61 |
-
prompt = gr.Textbox(placeholder="enter prompt", show_label=False)
|
| 62 |
video_inp = gr.Video(label="Video source", source="upload", type="filepath")
|
| 63 |
|
| 64 |
with gr.Column():
|
|
@@ -67,7 +104,7 @@ with gr.Blocks(css='style.css') as demo:
|
|
| 67 |
trim_in = gr.Slider(label="Cut video at (s)", minimun=2, maximum=10, step=1, value=2)
|
| 68 |
video_out = gr.Video(label="Pix2pix video result")
|
| 69 |
submit_btn = gr.Button("Generate Pix2Pix video")
|
| 70 |
-
|
| 71 |
inputs = [prompt,video_inp,seed_inp, trim_in]
|
| 72 |
outputs = [video_out]
|
| 73 |
|
|
|
|
| 9 |
|
| 10 |
def get_frames(video_in):
|
| 11 |
frames = []
|
| 12 |
+
#resize the video
|
| 13 |
+
clip = mp.VideoFileClip(video_in)
|
| 14 |
+
clip_resized = clip.resize(height=512)
|
| 15 |
+
clip_resized.write_videofile("video_resized.mp4")
|
| 16 |
+
|
| 17 |
+
# Opens the Video file with CV2
|
| 18 |
+
cap= cv2.VideoCapture("video_resized.mp4")
|
| 19 |
+
|
| 20 |
fps = cap.get(cv2.CAP_PROP_FPS)
|
| 21 |
i=0
|
| 22 |
while(cap.isOpened()):
|
|
|
|
| 59 |
|
| 60 |
return final_vid
|
| 61 |
|
| 62 |
+
title = """
|
| 63 |
+
<div style="text-align: center; max-width: 700px; margin: 0 auto;">
|
| 64 |
+
<div
|
| 65 |
+
style="
|
| 66 |
+
display: inline-flex;
|
| 67 |
+
align-items: center;
|
| 68 |
+
gap: 0.8rem;
|
| 69 |
+
font-size: 1.75rem;
|
| 70 |
+
"
|
| 71 |
+
>
|
| 72 |
+
<h1 style="font-weight: 900; margin-bottom: 7px; margin-top: 5px;">
|
| 73 |
+
Pix2Pix Video
|
| 74 |
+
</h1>
|
| 75 |
+
</div>
|
| 76 |
+
<p style="margin-bottom: 10px; font-size: 94%">
|
| 77 |
+
Apply Instruct Pix2Pix Diffusion to a video
|
| 78 |
+
</p>
|
| 79 |
+
</div>
|
| 80 |
+
"""
|
| 81 |
+
|
| 82 |
+
article = """
|
| 83 |
+
|
| 84 |
+
<div class="footer">
|
| 85 |
+
<p>
|
| 86 |
+
|
| 87 |
+
Follow <a href="https://twitter.com/fffiloni" target="_blank">Sylvain Filoni</a> for future updates 🤗
|
| 88 |
+
</p>
|
| 89 |
+
</div>
|
| 90 |
+
|
| 91 |
+
"""
|
| 92 |
|
| 93 |
with gr.Blocks(css='style.css') as demo:
|
| 94 |
with gr.Column(elem_id="col-container"):
|
| 95 |
+
gr.HTML(title)
|
| 96 |
with gr.Row():
|
| 97 |
with gr.Column():
|
| 98 |
+
prompt = gr.Textbox(label="Prompt", placeholder="enter prompt", show_label=False)
|
| 99 |
video_inp = gr.Video(label="Video source", source="upload", type="filepath")
|
| 100 |
|
| 101 |
with gr.Column():
|
|
|
|
| 104 |
trim_in = gr.Slider(label="Cut video at (s)", minimun=2, maximum=10, step=1, value=2)
|
| 105 |
video_out = gr.Video(label="Pix2pix video result")
|
| 106 |
submit_btn = gr.Button("Generate Pix2Pix video")
|
| 107 |
+
gr.HTML(article)
|
| 108 |
inputs = [prompt,video_inp,seed_inp, trim_in]
|
| 109 |
outputs = [video_out]
|
| 110 |
|