Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|