pujanpaudel commited on
Commit
21042da
·
verified ·
1 Parent(s): 5026e83

Rename inference.py to app.py

Browse files
Files changed (1) hide show
  1. inference.py → app.py +2 -2
inference.py → app.py RENAMED
@@ -93,8 +93,8 @@ def predict_distractor(model:DistractorTrained,tokenizer:T5Tokenizer,
93
  def main(user_input):
94
  device = "cuda" if torch.cuda.is_available() else "cpu"
95
  tokenizer_path:str = "./t5_tokenizer"
96
- qa_model_path:str = "./qa_trained_model/qa-t5-small.ckpt"
97
- distractor_model_path:str = "./distractor_trained_model/distractor_t5-small.ckpt"
98
  tokenizer = load_tokenizer(tokenizer_path)
99
  qa_model = load_qa_model(qa_model_path)
100
  distractor_model = load_distractor_model(distractor_model_path)
 
93
  def main(user_input):
94
  device = "cuda" if torch.cuda.is_available() else "cpu"
95
  tokenizer_path:str = "./t5_tokenizer"
96
+ qa_model_path:str = "./qa-t5-small.ckpt"
97
+ distractor_model_path:str = "./distractor_t5-small.ckpt"
98
  tokenizer = load_tokenizer(tokenizer_path)
99
  qa_model = load_qa_model(qa_model_path)
100
  distractor_model = load_distractor_model(distractor_model_path)