Spaces:
Runtime error
Runtime error
Commit ·
1c4dfa7
1
Parent(s): b5ca2ed
back to zero
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import os
|
|
| 5 |
import cv2
|
| 6 |
import uuid
|
| 7 |
import time
|
| 8 |
-
|
| 9 |
import subprocess
|
| 10 |
import matplotlib
|
| 11 |
matplotlib.use('Agg')
|
|
@@ -51,7 +51,7 @@ def sigmoid(x):
|
|
| 51 |
return 1 / (1 + np.exp(-x))
|
| 52 |
|
| 53 |
|
| 54 |
-
|
| 55 |
def inference(x, count_only_api, api_key,
|
| 56 |
img_size=288, seq_len=64, stride_length=32, stride_pad=3, batch_size=4,
|
| 57 |
miss_threshold=0.8, marks_threshold=0.6, median_pred_filter=True, center_crop=True, both_feet=True,
|
|
|
|
| 5 |
import cv2
|
| 6 |
import uuid
|
| 7 |
import time
|
| 8 |
+
import spaces
|
| 9 |
import subprocess
|
| 10 |
import matplotlib
|
| 11 |
matplotlib.use('Agg')
|
|
|
|
| 51 |
return 1 / (1 + np.exp(-x))
|
| 52 |
|
| 53 |
|
| 54 |
+
@spaces.GPU()
|
| 55 |
def inference(x, count_only_api, api_key,
|
| 56 |
img_size=288, seq_len=64, stride_length=32, stride_pad=3, batch_size=4,
|
| 57 |
miss_threshold=0.8, marks_threshold=0.6, median_pred_filter=True, center_crop=True, both_feet=True,
|