votuongquan2004@gmail.com commited on
Commit
1baa500
1 Parent(s): 5362fbc

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.onnx')
26
 
27
  # Load id-to-gloss mapping.
28
  id2gloss = pd.read_csv('gloss.csv', names=['id', 'gloss']).to_dict()['gloss']
@@ -82,7 +82,7 @@ def inference(
82
  output_message += f'Data processing time: {data_time:.2f} seconds\n'
83
  output_message += f'Model inference time: {model_time:.2f} seconds\n'
84
  output_message += f'Total time: {data_time + model_time:.2f} seconds'
85
- output_message += f'Input shape: {inputs.shape}\n'
86
 
87
  progress(1/2, desc='Completed')
88
 
 
22
  examples = []
23
 
24
  # Load the configuration file.
25
+ ort_session = ort.InferenceSession('VSL_SAM_SLR_V2_joint.onnx')
26
 
27
  # Load id-to-gloss mapping.
28
  id2gloss = pd.read_csv('gloss.csv', names=['id', 'gloss']).to_dict()['gloss']
 
82
  output_message += f'Data processing time: {data_time:.2f} seconds\n'
83
  output_message += f'Model inference time: {model_time:.2f} seconds\n'
84
  output_message += f'Total time: {data_time + model_time:.2f} seconds'
85
+ output_message += f'\nInput shape: {inputs.shape}'
86
 
87
  progress(1/2, desc='Completed')
88