Commit ·
f80d9b2
1
Parent(s): a3aba73
revert to lightning
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ def sigmoid(x):
|
|
| 50 |
return 1 / (1 + np.exp(-x))
|
| 51 |
|
| 52 |
|
| 53 |
-
def inference(x, count_only_api, api_key, img_size=
|
| 54 |
print(x)
|
| 55 |
#api = HfApi(token=os.environ['DATASET_SECRET'])
|
| 56 |
#out_file = str(uuid.uuid1())
|
|
|
|
| 50 |
return 1 / (1 + np.exp(-x))
|
| 51 |
|
| 52 |
|
| 53 |
+
def inference(x, count_only_api, api_key, img_size=192, 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):
|
| 54 |
print(x)
|
| 55 |
#api = HfApi(token=os.environ['DATASET_SECRET'])
|
| 56 |
#out_file = str(uuid.uuid1())
|