Spaces:

thoristhor
/

nasilemak

Build error

thoristhor commited on Sep 27, 2022

Commit

dca4cea

1 Parent(s): c121353

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def load_model():
     ## summary_mod_name = os.environ["summary_mod_name"]
     ## question_mod_name = os.environ["question_mod_name"]
     summary_mod_name = "mrm8488/t5-base-finetuned-summarize-news"
-    question_mod_name = "mrm8488/t5-base-finetuned-quartz"
     summary_model = T5ForConditionalGeneration.from_pretrained(summary_mod_name)
     summary_tokenizer = T5Tokenizer.from_pretrained(summary_mod_name)
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

     ## summary_mod_name = os.environ["summary_mod_name"]
     ## question_mod_name = os.environ["question_mod_name"]
     summary_mod_name = "mrm8488/t5-base-finetuned-summarize-news"
+    question_mod_name = "mrm8488/t5-base-finetuned-question-generation-ap"
     summary_model = T5ForConditionalGeneration.from_pretrained(summary_mod_name)
     summary_tokenizer = T5Tokenizer.from_pretrained(summary_mod_name)
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")