Hugging Face's logo

Spaces:
Saibo-creator
/
transformers-CFG-JSON-demo
Running

App Files Community
Saibo commited on
Commit
d51d76f
·
verified ·
1 Parent(s): 6ef22f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  from transformers_cfg.grammar_utils import IncrementalGrammarConstraint
6
  from transformers_cfg.generation.logits_process import GrammarConstrainedLogitsProcessor
7
 
8
- MODEL_NAME = "gpt2-medium"
9
 
10
  if __name__ == "__main__":
11
  # Define your model and your tokenizer
 
5
  from transformers_cfg.grammar_utils import IncrementalGrammarConstraint
6
  from transformers_cfg.generation.logits_process import GrammarConstrainedLogitsProcessor
7
 
8
+ MODEL_NAME = "gpt2"
9
 
10
  if __name__ == "__main__":
11
  # Define your model and your tokenizer