Spaces:

jeyasee
/

jlanguage-detector

Sleeping

jeyasee commited on Jul 30

Commit

7129900

verified ·

1 Parent(s): eb0b20b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ def detect_language(sentence):
     output = model(**tokenized_sentence)
     predictions = torch.nn.functional.softmax(output.logits, dim=-1)
     _, preds = torch.max(predictions, dim=-1)
-    return LANGUANGE_MAP[preds.item()]
     #return LANGUANGE_MAP[preds.item()]
 examples = [

     output = model(**tokenized_sentence)
     predictions = torch.nn.functional.softmax(output.logits, dim=-1)
     _, preds = torch.max(predictions, dim=-1)
+    return preds.item()
     #return LANGUANGE_MAP[preds.item()]
 examples = [