Spaces:
Sleeping
Sleeping
Rename inference.py to app.py
Browse files- 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 = "./
|
97 |
-
distractor_model_path:str = "./
|
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)
|