vpcom commited on
Commit
af4a236
·
1 Parent(s): 99388f0

test: let's see if we can use the additional_inputs properties

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -146,7 +146,7 @@ def evolve_text(textbox):
146
 
147
  possible_generation = "".join([x for x in generate(
148
  textbox, "", "<|endoftext|>",
149
- temperature=0.9, max_new_tokens=10,
150
  top_p=0.95, repetition_penalty=1.2,
151
  seed=42,
152
  )][-1]) #TODO: need a smarter way to do this
 
146
 
147
  possible_generation = "".join([x for x in generate(
148
  textbox, "", "<|endoftext|>",
149
+ temperature=additional_inputs[1].value, max_new_tokens=10,
150
  top_p=0.95, repetition_penalty=1.2,
151
  seed=42,
152
  )][-1]) #TODO: need a smarter way to do this