Upload app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import spaces
-from torch import exp
+import math
 from inflections_funcs import start_model, make_beams, get_beam_tokens, calculate_score_vectors
 
 # Global model initialization
@@ -14,7 +14,7 @@ def generate_beam_html(index, all_beams_data, dark_mode):
     """
     beam_tokens = all_beams_data["beam_tokens"][index]
     beam_scores = all_beams_data["score_vectors"][index]
-    beam_overall_score = exp(all_beams_data["sequences_scores"][index]).item()
+    beam_overall_score = math.exp(all_beams_data["sequences_scores"][index])
 
     # Construct Main HTML output
     main_container_style = "border: 2px solid lightblue; padding: 15px; border-radius: 8px; font-family: sans-serif;"