Spaces:
Sleeping
Sleeping
Sudoaptinstallpy3
commited on
Commit
•
44b4b17
1
Parent(s):
b5f6b10
Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
from mario_gpt import MarioLM, SampleOutput
|
3 |
-
import torch
|
4 |
-
from PIL import Image
|
5 |
|
6 |
# Load the Mario GPT model
|
7 |
-
|
8 |
-
mario_lm = MarioLM.from_pretrained(model_name)
|
9 |
|
10 |
# Define the function to generate Mario levels
|
11 |
-
def generate_mario_level(prompt, num_steps=
|
12 |
generated_level = mario_lm.sample(
|
13 |
prompts=[prompt],
|
14 |
num_steps=num_steps,
|
|
|
1 |
import gradio as gr
|
2 |
from mario_gpt import MarioLM, SampleOutput
|
|
|
|
|
3 |
|
4 |
# Load the Mario GPT model
|
5 |
+
mario_lm = MarioLM()
|
|
|
6 |
|
7 |
# Define the function to generate Mario levels
|
8 |
+
def generate_mario_level(prompt, num_steps=1400, temperature=2.0):
|
9 |
generated_level = mario_lm.sample(
|
10 |
prompts=[prompt],
|
11 |
num_steps=num_steps,
|