harithapliyal commited on
Commit
c4f0d2a
1 Parent(s): ae13b1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = '/content/drive/MyDrive/HBQA/t5small-30epoch'
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)