Spaces:
Running
Running
Pratyush Maini
commited on
Commit
·
dffff5c
1
Parent(s):
2c53d28
strong
Browse files
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):
|