Commit
·
c45074e
1
Parent(s):
63e4986
Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ data_path = 'signs'
|
|
51 |
# Custom CSS to control output video size
|
52 |
custom_css = """
|
53 |
#landmarked_video {
|
54 |
-
max-height:
|
55 |
width: auto;
|
56 |
}
|
57 |
"""
|
@@ -245,7 +245,7 @@ def prod_function(model_pretrained, prod_ds):
|
|
245 |
return prod_pred
|
246 |
|
247 |
# Function to get landmarked video
|
248 |
-
def save_video_to_mp4(video_tensor, fps=
|
249 |
# Convert pytorch tensor to numpy ndarray
|
250 |
video_numpy = video_tensor.permute(0, 2, 3, 1).cpu().numpy()
|
251 |
# Normalize values to [0, 255] if necessary
|
|
|
51 |
# Custom CSS to control output video size
|
52 |
custom_css = """
|
53 |
#landmarked_video {
|
54 |
+
max-height: 600px;
|
55 |
width: auto;
|
56 |
}
|
57 |
"""
|
|
|
245 |
return prod_pred
|
246 |
|
247 |
# Function to get landmarked video
|
248 |
+
def save_video_to_mp4(video_tensor, fps=15):
|
249 |
# Convert pytorch tensor to numpy ndarray
|
250 |
video_numpy = video_tensor.permute(0, 2, 3, 1).cpu().numpy()
|
251 |
# Normalize values to [0, 255] if necessary
|