Spaces:
Sleeping
Sleeping
Sudoaptinstallpy3
commited on
Commit
•
8fb7ddc
1
Parent(s):
44b4b17
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from mario_gpt import MarioLM, SampleOutput
|
|
5 |
mario_lm = MarioLM()
|
6 |
|
7 |
# Define the function to generate Mario levels
|
8 |
-
def generate_mario_level(prompt, num_steps=
|
9 |
generated_level = mario_lm.sample(
|
10 |
prompts=[prompt],
|
11 |
num_steps=num_steps,
|
|
|
5 |
mario_lm = MarioLM()
|
6 |
|
7 |
# Define the function to generate Mario levels
|
8 |
+
def generate_mario_level(prompt, num_steps=1000, temperature=2.0):
|
9 |
generated_level = mario_lm.sample(
|
10 |
prompts=[prompt],
|
11 |
num_steps=num_steps,
|