thoristhor commited on
Commit
dca4cea
1 Parent(s): c121353

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def load_model():
29
  ## summary_mod_name = os.environ["summary_mod_name"]
30
  ## question_mod_name = os.environ["question_mod_name"]
31
  summary_mod_name = "mrm8488/t5-base-finetuned-summarize-news"
32
- question_mod_name = "mrm8488/t5-base-finetuned-quartz"
33
  summary_model = T5ForConditionalGeneration.from_pretrained(summary_mod_name)
34
  summary_tokenizer = T5Tokenizer.from_pretrained(summary_mod_name)
35
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
29
  ## summary_mod_name = os.environ["summary_mod_name"]
30
  ## question_mod_name = os.environ["question_mod_name"]
31
  summary_mod_name = "mrm8488/t5-base-finetuned-summarize-news"
32
+ question_mod_name = "mrm8488/t5-base-finetuned-question-generation-ap"
33
  summary_model = T5ForConditionalGeneration.from_pretrained(summary_mod_name)
34
  summary_tokenizer = T5Tokenizer.from_pretrained(summary_mod_name)
35
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")