add zerogpu

app.py

@@ -5,6 +5,7 @@ import os
 import cv2
 import uuid
 import time
+import spaces
 import subprocess
 import matplotlib
 matplotlib.use('Agg')
@@ -59,6 +60,7 @@ def sigmoid(x):
     return 1 / (1 + np.exp(-x))
 
 
+@spaces.GPU(duration=180)
 def inference(x, count_only_api, api_key, img_size=224, seq_len=64, stride_length=32, stride_pad=3, batch_size=4, miss_threshold=0.8, marks_threshold=0.6, median_pred_filter=True, center_crop=True, both_feet=True, api_call=False):
     print(x)
     #api = HfApi(token=os.environ['DATASET_SECRET'])