Update app.py

app.py

@@ -3,6 +3,7 @@ import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 import pandas as pd
+import math
 
 # Load model and tokenizer
 model_ids = {
@@ -52,7 +53,7 @@ def compare_models(text, top_k=5):
             topk_ids = topk.indices.tolist()
             topk_scores = topk.values.tolist()
             topk_tokens = tokenizer.convert_ids_to_tokens(topk_ids)
-            topk_probs = [round(float(torch.exp(s)), 4) for s in topk_scores]
+            topk_probs = [round(math.exp(s), 4) for s in topk_scores]
             pair_list = [f"{tok} ({prob})" for tok, prob in zip(topk_tokens, topk_probs)]
             topk_list.append(", ".join(pair_list))