makanaan commited on
Commit
e093731
·
1 Parent(s): 2064b17

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,9 +5,9 @@ from transformers import AutoTokenizer, AutoModelWithLMHead
5
  #from os import path
6
 
7
  #MODEL_DIRECTORY = "/models/mrm8488-t5-base-finetuned-emotion"
8
- tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion", use_fast=False)
9
  #if not path.exists(MODEL_DIRECTORY):
10
- model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
11
  # model.save_pretrained(MODEL_DIRECTORY)
12
  #else:
13
  # model = AutoModelWithLMHead.from_pretrained(MODEL_DIRECTORY)
 
5
  #from os import path
6
 
7
  #MODEL_DIRECTORY = "/models/mrm8488-t5-base-finetuned-emotion"
8
+ tokenizer = AutoTokenizer.from_pretrained("tuner007/pegasus_paraphrase", use_fast=False)
9
  #if not path.exists(MODEL_DIRECTORY):
10
+ model = AutoModelWithLMHead.from_pretrained("tuner007/pegasus_paraphrase")
11
  # model.save_pretrained(MODEL_DIRECTORY)
12
  #else:
13
  # model = AutoModelWithLMHead.from_pretrained(MODEL_DIRECTORY)