arjunpatel commited on
Commit
cc121a0
·
1 Parent(s): 20862de

Formatting changes

Browse files

Thanks to macklon for pointing this out! Added json dumps to parameters to clear up issue there, and I removed a sentence from move desc to assist formatting as well.

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -4,6 +4,7 @@ from transformers import pipeline
4
  from utils import format_moves
5
  import pandas as pd
6
  import tensorflow as tf
 
7
 
8
  model_checkpoint = "distilgpt2"
9
 
@@ -19,10 +20,14 @@ tf.random.set_seed(0)
19
 
20
 
21
  def update_history(df, move_name, move_desc, generation, parameters):
 
 
 
 
22
  new_row = [{"Move Name": move_name,
23
  "Move Description": move_desc,
24
  "Generation Type": generation,
25
- "Parameters": parameters}]
26
  return pd.concat([df, pd.DataFrame(new_row)])
27
 
28
 
 
4
  from utils import format_moves
5
  import pandas as pd
6
  import tensorflow as tf
7
+ import json
8
 
9
  model_checkpoint = "distilgpt2"
10
 
 
20
 
21
 
22
  def update_history(df, move_name, move_desc, generation, parameters):
23
+ # get rid of first seed phrase
24
+
25
+ move_desc = move_desc.split("\n")[1:]
26
+ move_desc = "\n".join(move_desc)
27
  new_row = [{"Move Name": move_name,
28
  "Move Description": move_desc,
29
  "Generation Type": generation,
30
+ "Parameters": json.dumps(parameters)}]
31
  return pd.concat([df, pd.DataFrame(new_row)])
32
 
33