nicolasbuitragob commited on
Commit
1988c5a
·
1 Parent(s): cad429f

edits in filepath

Browse files
Files changed (3) hide show
  1. .gitignore +4 -0
  2. tasks.py +1 -1
  3. vitpose.py +1 -1
.gitignore CHANGED
@@ -172,3 +172,7 @@ cython_debug/
172
 
173
  # PyPI configuration file
174
  .pypirc
 
 
 
 
 
172
 
173
  # PyPI configuration file
174
  .pypirc
175
+
176
+ static/
177
+ *.mp4
178
+ *.MOV
tasks.py CHANGED
@@ -13,7 +13,7 @@ def process_video(video_path: str,vitpose: VitPose,user_id: str):
13
  logger.info(f"starting task {video_path}")
14
 
15
 
16
- new_file_name = video_path.split(".")[0] + "edited." + video_path.split(".")[1]
17
  new_file_name = os.path.join("static", new_file_name)
18
  logger.info(f"new file name {new_file_name}")
19
 
 
13
  logger.info(f"starting task {video_path}")
14
 
15
 
16
+ new_file_name = video_path.split(".")[0] + '.' + video_path.split(".")[1]
17
  new_file_name = os.path.join("static", new_file_name)
18
  logger.info(f"new file name {new_file_name}")
19
 
vitpose.py CHANGED
@@ -38,7 +38,7 @@ class VitPose:
38
  if not ret:
39
  break
40
  frames.append(frame)
41
- return frames
42
 
43
  def run(self,video):
44
  frames = self.video_to_frames(video)
 
38
  if not ret:
39
  break
40
  frames.append(frame)
41
+ return frames[:2]
42
 
43
  def run(self,video):
44
  frames = self.video_to_frames(video)