Spaces:
Build error
Build error
Commit
·
459b405
1
Parent(s):
4e40787
Update app.py
Browse files
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 = "t5-small"
|
32 |
-
|
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 = "t5-small"
|
32 |
+
question_mod_name= "t5-small"
|
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")
|