ethanlshen commited on
Commit
09cab71
1 Parent(s): f451377

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def decode(tokenizer, encoding):
63
  encoding = encoding[:eos_locs[0]]
64
  return tokenizer.decode(encoding.to(torch.int32).tolist())
65
 
66
- @spaces.GPU(duration=3)
67
  def update_options(input, num_tokens):
68
  tokenized_prompts = tokenizer.encode([input], True, False)
69
  print("Processed prompt")
 
63
  encoding = encoding[:eos_locs[0]]
64
  return tokenizer.decode(encoding.to(torch.int32).tolist())
65
 
66
+ @spaces.GPU
67
  def update_options(input, num_tokens):
68
  tokenized_prompts = tokenizer.encode([input], True, False)
69
  print("Processed prompt")