Commit
·
ad5aa37
1
Parent(s):
2204841
Update app.py
Browse files
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 |
-
|
263 |
|
264 |
-
for val in np.array(predicted_prod_label.cpu().numpy()):
|
265 |
-
|
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 |
|