Spaces:
Sleeping
Sleeping
GradApplicationDocuments
commited on
Commit
•
4a5c5be
1
Parent(s):
d472283
Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ def main():
|
|
92 |
elif option == "BERT Revised":
|
93 |
model = HF_BertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/BERTGradRevised").to(device)
|
94 |
elif option == "D-BERT Generated":
|
95 |
-
model =
|
96 |
else:
|
97 |
model = HF_DistilBertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/DistilBERTGradRevised").to(device)
|
98 |
return model
|
@@ -123,7 +123,7 @@ def main():
|
|
123 |
|
124 |
|
125 |
# Load the selected trained model
|
126 |
-
if option in ("BERT
|
127 |
tokenizer = load_tokenizer(option)
|
128 |
model = load_model(option)
|
129 |
else:
|
|
|
92 |
elif option == "BERT Revised":
|
93 |
model = HF_BertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/BERTGradRevised").to(device)
|
94 |
elif option == "D-BERT Generated":
|
95 |
+
model = HF_DistilBertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/DistilBERTGradGen").to(device)
|
96 |
else:
|
97 |
model = HF_DistilBertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/DistilBERTGradRevised").to(device)
|
98 |
return model
|
|
|
123 |
|
124 |
|
125 |
# Load the selected trained model
|
126 |
+
if option in ("BERT Generated", "BERT Revised", "D-BERT Generated", "D-BERT Revised"):
|
127 |
tokenizer = load_tokenizer(option)
|
128 |
model = load_model(option)
|
129 |
else:
|