Update app.py

app.py

@@ -65,7 +65,7 @@ def ensure_float(value):
     return None
 
 # Function to process and visualize log probs with multiple analyses
-def visualize_logprobs(json_input, prob_filter=-float('inf')):
+def visualize_logprobs(json_input, prob_filter=-1e9):  # Changed -float('inf') to -1e9
     try:
         # Parse the input (handles both JSON and Python dictionaries)
         data = parse_input(json_input)
@@ -113,7 +113,7 @@ def visualize_logprobs(json_input, prob_filter=-float('inf')):
 
         # If no valid data after filtering, return error messages
         if not logprobs:
-            return "No finite log probabilities to visualize after filtering.", None, None, None, None, None, None, None, None, None, None
+            return ("No finite log probabilities to visualize after filtering.", None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None)
 
         # 1. Main Log Probability Plot (with click for tokens)
         fig_main, ax_main = plt.subplots(figsize=(10, 5))
@@ -534,7 +534,7 @@ with gr.Blocks(title="Log Probability Visualizer") as app:
             lines=10,
             placeholder="Paste your JSON (e.g., {\"content\": [...]}) or Python dict (e.g., {'content': [...]}) here...",
         )
-        prob_filter = gr.Slider(minimum=-float('inf'), maximum=0, value=-float('inf'), label="Log Probability Filter (≥)")
+        prob_filter = gr.Slider(minimum=-1e9, maximum=0, value=-1e9, label="Log Probability Filter (≥)")  # Changed -float('inf') to -1e9
 
     with gr.Row():
         plot_output = gr.HTML(label="Log Probability Plot (Click for Tokens)")