Kaushik066 commited on
Commit
ad5aa37
·
1 Parent(s): 2204841

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -259,10 +259,10 @@ def translate_sign_language(gesture):
259
 
260
  idx_to_label = model_pretrained.config.id2label
261
  #gesture_translation = predicted_prod_label.cpu().numpy()
262
- #gesture_translation = idx_to_label[predicted_prod_label.cpu().numpy()]
263
 
264
- for val in np.array(predicted_prod_label.cpu().numpy()):
265
- gesture_translation = idx_to_label[val]
266
 
267
  return gesture_translation
268
 
 
259
 
260
  idx_to_label = model_pretrained.config.id2label
261
  #gesture_translation = predicted_prod_label.cpu().numpy()
262
+ gesture_translation = idx_to_label[predicted_prod_label.cpu().numpy()]
263
 
264
+ #for val in np.array(predicted_prod_label.cpu().numpy()):
265
+ # gesture_translation = idx_to_label[val]
266
 
267
  return gesture_translation
268