Spaces:
Sleeping
Sleeping
[email protected]
commited on
Commit
·
7ec94e0
1
Parent(s):
0fd4dc5
test
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ description = '''
|
|
22 |
examples = []
|
23 |
|
24 |
# Load the configuration file.
|
25 |
-
ort_session = ort.InferenceSession('
|
26 |
|
27 |
# Load id-to-gloss mapping.
|
28 |
id2gloss = pd.read_csv('gloss.csv', names=['id', 'gloss']).to_dict()['gloss']
|
@@ -70,6 +70,8 @@ def inference(
|
|
70 |
)
|
71 |
end_time = time.time()
|
72 |
model_time = end_time - start_time
|
|
|
|
|
73 |
|
74 |
if len(predictions) == 0:
|
75 |
output_message = 'No sign language detected in the video. Please try again.'
|
|
|
22 |
examples = []
|
23 |
|
24 |
# Load the configuration file.
|
25 |
+
ort_session = ort.InferenceSession('VSL_SAM_SLR_V2_bone.onnx')
|
26 |
|
27 |
# Load id-to-gloss mapping.
|
28 |
id2gloss = pd.read_csv('gloss.csv', names=['id', 'gloss']).to_dict()['gloss']
|
|
|
70 |
)
|
71 |
end_time = time.time()
|
72 |
model_time = end_time - start_time
|
73 |
+
|
74 |
+
output_message += f'Input shape: {inputs.shape}\n'
|
75 |
|
76 |
if len(predictions) == 0:
|
77 |
output_message = 'No sign language detected in the video. Please try again.'
|