Spaces:

Kaushik066
/

indian_sign_language_translation

Running

Kaushik066 commited on Feb 13

Commit

15be605

1 Parent(s): 0d07950

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -255,7 +255,7 @@ def translate_sign_language(gesture):
     predicted_prod_label = prod_function(model_pretrained, prod_ds)
     # Identify the hand gesture
-    predicted_prod_label = predicted_prod_label.squeeze(1)
     idx_to_label = model_pretrained.config.id2label
     gesture_translation = idx_to_label[np.array(predicted_prod_label)]

     predicted_prod_label = prod_function(model_pretrained, prod_ds)
     # Identify the hand gesture
+    predicted_prod_label = predicted_prod_label.squeeze(0)
     idx_to_label = model_pretrained.config.id2label
     gesture_translation = idx_to_label[np.array(predicted_prod_label)]