Update app.py

app.py

@@ -5,10 +5,10 @@ from PIL import Image
 
 # Load the Mario GPT model
 model_name = "shyamsn97/Mario-GPT2-700-context-length"
-mario_lm = MarioLM(lm_path=model_name, tokenizer_path=model_name)
+mario_lm = MarioLM.from_pretrained(model_name)
 
 # Define the function to generate Mario levels
-def generate_mario_level(prompt, num_steps=1400, temperature=2.0):
+def generate_mario_level(prompt, num_steps=1000, temperature=2.0):
     generated_level = mario_lm.sample(
         prompts=[prompt],
         num_steps=num_steps,