Update app.py

app.py

@@ -1,14 +1,11 @@
 import gradio as gr
 from mario_gpt import MarioLM, SampleOutput
-import torch
-from PIL import Image
 
 # Load the Mario GPT model
-model_name = "shyamsn97/Mario-GPT2-700-context-length"
-mario_lm = MarioLM.from_pretrained(model_name)
+mario_lm = MarioLM()
 
 # Define the function to generate Mario levels
-def generate_mario_level(prompt, num_steps=1000, temperature=2.0):
+def generate_mario_level(prompt, num_steps=1400, temperature=2.0):
     generated_level = mario_lm.sample(
         prompts=[prompt],
         num_steps=num_steps,