dylanplummer commited on
Commit
1c4dfa7
·
1 Parent(s): b5ca2ed

back to zero

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
  import cv2
6
  import uuid
7
  import time
8
- #import spaces
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
- #@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,
 
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,