fixed code
This commit is contained in:
@@ -1,9 +1,9 @@
|
||||
import torch
|
||||
|
||||
from app.ml.model_manager import model_manager
|
||||
from app.config.logging import logger
|
||||
from app.ml.prediction_result import PredictionResult
|
||||
|
||||
|
||||
class TextClassifier:
|
||||
def predict(self, text: str):
|
||||
inputs = model_manager.text_tokenizer(
|
||||
|
||||
Reference in New Issue
Block a user