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)
|