Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
@@ -6,6 +7,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 model = GPT2LMHeadModel.from_pretrained("gpt2")
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
+@spaces.GPU
 def get_next_token_probs(text):
     # Handle empty input
     if not text.strip():