Ahsen Khaliq commited on
Commit
9d828e1
·
1 Parent(s): 0f4ac7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -1,9 +1,15 @@
1
  import os
2
  os.system('gdown https://drive.google.com/uc?id=1hv0xwdbdf-Ym06Hpjy6wKyIklNmKmB99') # kapao_s_coco.pt
 
3
  import gradio as gr
 
 
4
 
5
  def greet(vid):
6
- os.system("python demos/general.py --weights kapao_s_coco.pt --start 0 --end 2 --gif --fps --vid "+vid)
 
 
 
7
  return 'flash_mob_inference_kapao_s_coco.gif'
8
 
9
  title = "Kapao"
 
1
  import os
2
  os.system('gdown https://drive.google.com/uc?id=1hv0xwdbdf-Ym06Hpjy6wKyIklNmKmB99') # kapao_s_coco.pt
3
+ os.system('pip install moviepy')
4
  import gradio as gr
5
+ import moviepy.editor as mp
6
+
7
 
8
  def greet(vid):
9
+ clip = mp.VideoFileClip(vid)
10
+ clip_resized = clip.resize(height=360)
11
+ clip_resized.write_videofile("movie_resized.mp4")
12
+ os.system("python demos/general.py --weights kapao_s_coco.pt --start 0 --end 2 --gif --fps --vid movie_resized.mp4")
13
  return 'flash_mob_inference_kapao_s_coco.gif'
14
 
15
  title = "Kapao"