Update app.py

app.py

@@ -11,7 +11,7 @@ device = torch.device('cuda')
 mario_lm = mario_lm.to(device)
 TILE_DIR = "data/tiles"
 
-def update(prompt, progress=gr.Progress(track_tqdm=True)):
+def update(prompt):
     prompts = [prompt]
     generated_level = mario_lm.sample(
         prompts=prompts,
@@ -33,6 +33,6 @@ with gr.Blocks() as demo:
         inputs=prompt,
         outputs=level_image,
         fn=update,
-        cache_examples=False,
+        cache_examples=True,
     )
 demo.launch()