Update app.py

app.py

@@ -1,4 +1,3 @@
-import spaces
 import gradio as gr
 import torch
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
@@ -7,7 +6,6 @@ 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():
@@ -48,10 +46,13 @@ with gr.Blocks(css="footer {display: none}") as demo:
     # Input textbox
     input_text = gr.Textbox(
         label="Text Input",
-        placeholder="Type here and watch predictions update...",
+        placeholder="Type text here...",
         value="The weather tomorrow will be"
     )
     
+    # Predict button
+    predict_btn = gr.Button("Predict Next Tokens")
+    
     # Simple header for results
     gr.Markdown("##### Most likely next tokens:")
     
@@ -64,8 +65,8 @@ with gr.Blocks(css="footer {display: none}") as demo:
     
     token_outputs = [token1, token2, token3, token4, token5]
     
-    # Set up the live update
-    input_text.change(
+    # Set up button click event
+    predict_btn.click(
         fn=get_next_token_probs,
         inputs=input_text,
         outputs=token_outputs