diff --git a/app.py b/app.py index 250c04e..ca60b2e 100644 --- a/app.py +++ b/app.py @@ -1,4 +1,3 @@ -import torch from flask import Flask, request import classla import gc @@ -7,8 +6,6 @@ classla.download('sl') classla.download('sl', type='standard_jos') classla.download('sl', type='nonstandard') -print(f'USE GPU = {torch.cuda.is_available()}') - nlp_standard_UD = classla.Pipeline('sl', pos_use_lexicon=True) nlp_standard_JOS = classla.Pipeline('sl', pos_use_lexicon=True, type='standard_jos') nlp_nonstandard_UD = classla.Pipeline('sl', type='nonstandard')