Spaces:

Kaushik066
/

indian_sign_language_translation

Running

Kaushik066 commited on Feb 13

Commit

9bc671f

1 Parent(s): ad5aa37

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -259,7 +259,7 @@ def translate_sign_language(gesture):
     idx_to_label = model_pretrained.config.id2label
     #gesture_translation = predicted_prod_label.cpu().numpy()
-    gesture_translation = idx_to_label[predicted_prod_label.cpu().numpy()]
     #for val in np.array(predicted_prod_label.cpu().numpy()):
     #  gesture_translation = idx_to_label[val]

     idx_to_label = model_pretrained.config.id2label
     #gesture_translation = predicted_prod_label.cpu().numpy()
+    gesture_translation = idx_to_label[predicted_prod_label.cpu().numpy().item()] # Convert to a scalar
     #for val in np.array(predicted_prod_label.cpu().numpy()):
     #  gesture_translation = idx_to_label[val]