Spaces:

davanstrien
/

next-token

Sleeping

davanstrien HF Staff commited on May 6

Commit

5a020b8

verified ·

1 Parent(s): 93b1c2c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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():

 import gradio as gr
 import torch
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 model = GPT2LMHeadModel.from_pretrained("gpt2")
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 def get_next_token_probs(text):
     # Handle empty input
     if not text.strip():