try diff method

app.py

@@ -33,6 +33,12 @@ def identify_language(txt):
   return languages_map[id2label[str(y_pred[0])]]
 
 
+def identify_language2(txt):
+    tokenized_txt = tokenizer(txt, return_tensors='pt')
+    output = model(**tokenized_txt)
+    predictions = torch.nn.functional.softmax(output.logits, dim=-1)
+    _, preds = torch.max(predictions, dim=-1)
+    return languages_map[id2labe[str(preds.item())]]
 
 #with gr.Row():
 examples = [
@@ -72,7 +78,7 @@ article = ('''## Suppoted Langauges \n
 
 
 gr.Interface(
-    fn=identify_language,
+    fn=identify_language2,
     inputs=inputs,
     outputs=outputs,
     verbose=True,