Pratyush Maini commited on
Commit
dffff5c
·
1 Parent(s): 2c53d28
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -12,11 +12,11 @@ def load_csv(model_name):
12
  def get_result(model_name, target_string):
13
  df = load_csv(model_name)
14
  row = df[df['target_str'] == target_string].iloc[0]
15
- num_free_tokens = row['num_free_tokens']
16
- target_length = row['target_length']
17
  optimal_prompt = row['optimal_prompt']
18
- ratio = row['ratio']
19
- memorized = row['memorized']
20
  return num_free_tokens, target_length, optimal_prompt, ratio, memorized
21
 
22
  def update_csv_dropdown(model_name):
 
12
  def get_result(model_name, target_string):
13
  df = load_csv(model_name)
14
  row = df[df['target_str'] == target_string].iloc[0]
15
+ num_free_tokens = int(row['num_free_tokens']) # Convert to regular int
16
+ target_length = int(row['target_length']) # Convert to regular int
17
  optimal_prompt = row['optimal_prompt']
18
+ ratio = float(row['ratio']) # Convert to regular float
19
+ memorized = bool(row['memorized']) # Convert to regular bool
20
  return num_free_tokens, target_length, optimal_prompt, ratio, memorized
21
 
22
  def update_csv_dropdown(model_name):