multimodalart HF Staff commited on
Commit
7cc62e7
·
verified ·
1 Parent(s): 9f829c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -51,11 +51,12 @@ def generate(pipes, enemies, blocks, elevation, temperature = 2.0, level_size =
51
  if prompt == "":
52
  prompt = f"{pipes} pipes, {enemies} enemies, {blocks} blocks, {elevation} elevation"
53
  print(f"Using prompt: {prompt}")
 
54
  prompts = [prompt]
55
  generated_level = mario_lm.sample(
56
  prompts=prompts,
57
  num_steps=level_size,
58
- temperature=temperature,
59
  use_tqdm=True
60
  )
61
  filename = make_html_file(generated_level)
 
51
  if prompt == "":
52
  prompt = f"{pipes} pipes, {enemies} enemies, {blocks} blocks, {elevation} elevation"
53
  print(f"Using prompt: {prompt}")
54
+ print(f"Using temperature: {temperature}")
55
  prompts = [prompt]
56
  generated_level = mario_lm.sample(
57
  prompts=prompts,
58
  num_steps=level_size,
59
+ temperature=float(temperature),
60
  use_tqdm=True
61
  )
62
  filename = make_html_file(generated_level)