Spaces:
Build error
Build error
Commit
·
ae1a1df
1
Parent(s):
1b989ca
Update app.py
Browse files
app.py
CHANGED
@@ -28,8 +28,8 @@ def load_model():
|
|
28 |
nltk.download('omw-1.4')
|
29 |
## summary_mod_name = os.environ["summary_mod_name"]
|
30 |
## question_mod_name = os.environ["question_mod_name"]
|
31 |
-
summary_mod_name =
|
32 |
-
question_mod_name =
|
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")
|
|
|
28 |
nltk.download('omw-1.4')
|
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")
|