Update app.py

app.py

@@ -1,7 +1,7 @@
-
 import gradio as gr
 import torch
 import uuid
+import spaces
 from mario_gpt.dataset import MarioDataset
 from mario_gpt.prompter import Prompter
 from mario_gpt.lm import MarioLM
@@ -46,6 +46,7 @@ def make_html_file(generated_level):
 </html>''')
     return f"demo-{unique_id}.html"
 
+@spaces.GPU
 def generate(pipes, enemies, blocks, elevation, temperature = 2.0, level_size = 1399, prompt = ""):
     if prompt == "":
         prompt = f"{pipes} pipes, {enemies} enemies, {blocks} blocks, {elevation} elevation"