votuongquan2004@gmail.com commited on
Commit
7ec94e0
1 Parent(s): 0fd4dc5
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -22,7 +22,7 @@ description = '''
22
  examples = []
23
 
24
  # Load the configuration file.
25
- ort_session = ort.InferenceSession('VSL_SAM_SLR_V2_bone_motion.onnx')
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.'