terryyz commited on
Commit
bc55571
Β·
1 Parent(s): 275e3c8
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -2042,10 +2042,17 @@ def build_ui():
2042
  state0, state1, vote_type
2043
  )
2044
 
 
 
 
 
2045
  # Always show thank you message and clear everything immediately
2046
  gr.Info("Thank you for your vote! πŸŽ‰ Your feedback has been recorded and new models have been selected.", duration=5)
 
 
 
2047
 
2048
- # Get new random models
2049
  model_a, model_b = get_random_models()
2050
 
2051
  # Clear everything and start fresh immediately, but preserve examples
 
2042
  state0, state1, vote_type
2043
  )
2044
 
2045
+ # Get the model names from the current session
2046
+ model_a = state0["model_name"] if state0 else "Unknown"
2047
+ model_b = state1["model_name"] if state1 else "Unknown"
2048
+
2049
  # Always show thank you message and clear everything immediately
2050
  gr.Info("Thank you for your vote! πŸŽ‰ Your feedback has been recorded and new models have been selected.", duration=5)
2051
+
2052
+ # revval the model names in the info message
2053
+ gr.Info(f"Now you can see model names! πŸ‘€ \nModel A: {model_a}, Model B: {model_b}", duration=15)
2054
 
2055
+ # Get new random models for the next session
2056
  model_a, model_b = get_random_models()
2057
 
2058
  # Clear everything and start fresh immediately, but preserve examples