harithapliyal
commited on
Commit
•
c4f0d2a
1
Parent(s):
ae13b1c
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ print (predicted_Docid)
|
|
55 |
|
56 |
# @title Load Prediction Model
|
57 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
58 |
-
MODEL_FOLDER = '
|
59 |
|
60 |
# Load the corresponding tokenizer
|
61 |
tokenizer = T5Tokenizer.from_pretrained( MODEL_FOLDER)
|
|
|
55 |
|
56 |
# @title Load Prediction Model
|
57 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
58 |
+
MODEL_FOLDER = 't5small-30epoch'
|
59 |
|
60 |
# Load the corresponding tokenizer
|
61 |
tokenizer = T5Tokenizer.from_pretrained( MODEL_FOLDER)
|