Upload app.py

app.py

@@ -74,10 +74,11 @@ def predict(prompt, dark_mode):
         return "<span style='color: grey'>No tokens generated.</span>", "", gr.update(visible=False), gr.update(visible=False), None, 0
 
     # Initialize state with all beam data
+    # Convert tensors to lists to avoid ZeroGPU serialization errors
     all_beams_data = {
         "beam_tokens": beam_tokens,
         "score_vectors": score_vectors,
-        "sequences_scores": generated_dicts.sequences_scores
+        "sequences_scores": generated_dicts.sequences_scores.tolist() if hasattr(generated_dicts.sequences_scores, 'tolist') else generated_dicts.sequences_scores
     }
 
     # Generate HTML for the first beam (index 0)