set self.metric

pipeline.py

@@ -6,7 +6,7 @@ import json
 import numpy as np
 from fastai.learner import load_learner
 
-"""
+
 class OrdinalRegressionMetric(Metric):
     def __init__(self):
         super().__init__()
@@ -46,7 +46,6 @@ class OrdinalRegressionMetric(Metric):
         # Calculate the normalized metric value
         metric_value = 1/(self.total / self.count)
         return metric_value
-"""
 
 class PreTrainedPipeline():
     def __init__(self, path=""):
@@ -54,6 +53,7 @@ class PreTrainedPipeline():
         # Preload all the elements you are going to need at inference.
         # For instance your model, processors, tokenizer that might be needed.
         # This function is only called once, so do all the heavy processing I/O here"""
+        self.metric = OrdinalRegressionMetric()
         self.model = load_learner(os.path.join(path, "textfile3-2.pk1"))
         with open(os.path.join(path, "config.json")) as config:
             config = json.load(config)