Hugging Face's logo

Spaces:
appvoid
/
playground
Runtime error

App Files Community
appvoid commited on
Commit
1ec2891
·
verified ·
1 Parent(s): d754671

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
10
  tokenizer = AutoTokenizer.from_pretrained("gpt2")
11
  model = AutoModelForCausalLM.from_pretrained("gpt2")
12
 
13
- @space.GPU
14
  def text_generation(input_text, seed):
15
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
16
  torch.manual_seed(seed) # Max value: 18446744073709551615
 
10
  tokenizer = AutoTokenizer.from_pretrained("gpt2")
11
  model = AutoModelForCausalLM.from_pretrained("gpt2")
12
 
13
+ @spaces.GPU
14
  def text_generation(input_text, seed):
15
  input_ids = tokenizer(input_text, return_tensors="pt").input_ids
16
  torch.manual_seed(seed) # Max value: 18446744073709551615