Spaces:
Paused
Paused
Commit
·
1988c5a
1
Parent(s):
cad429f
edits in filepath
Browse files- .gitignore +4 -0
- tasks.py +1 -1
- 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] +
|
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)
|