dmariko commited on
Commit
c58b413
·
1 Parent(s): 4c507d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -22,13 +22,13 @@ import matplotlib.font_manager as fm
22
  MODEL_NAME = 'https://huggingface.co/yseop/FNP_T5_D2T_complete'
23
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
24
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
25
- config = AutoConfig.from_pretrained(MODEL_NAME)
26
 
27
  MODEL_BUF = {
28
  "name": MODEL_NAME,
29
  "tokenizer": tokenizer,
30
- "model": model,
31
- "config": config
32
  }
33
 
34
  font_dir = ['./']
@@ -44,7 +44,7 @@ def change_model_name(name):
44
  MODEL_BUF["name"] = name
45
  MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
46
  MODEL_BUF["model"] = AutoModelForSeq2SeqLM.from_pretrained(name)
47
- MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
48
 
49
 
50
  def generate(text, model, tokenizer):
 
22
  MODEL_NAME = 'https://huggingface.co/yseop/FNP_T5_D2T_complete'
23
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
24
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
25
+ #config = AutoConfig.from_pretrained(MODEL_NAME)
26
 
27
  MODEL_BUF = {
28
  "name": MODEL_NAME,
29
  "tokenizer": tokenizer,
30
+ "model": model
31
+ #"config": config
32
  }
33
 
34
  font_dir = ['./']
 
44
  MODEL_BUF["name"] = name
45
  MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
46
  MODEL_BUF["model"] = AutoModelForSeq2SeqLM.from_pretrained(name)
47
+ #MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
48
 
49
 
50
  def generate(text, model, tokenizer):