Ritvik19 commited on
Commit
e63e136
·
1 Parent(s): 3d82952

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -2
pipeline.py CHANGED
@@ -1,13 +1,12 @@
1
  import cleantext
2
  import joblib
3
- from huggingface_hub import hf_hub_download
4
  import os
5
 
6
  class PreTrainedPipeline():
7
  def __init__(self, path) -> None:
8
  self.models = self.load_models(path)
9
 
10
- def load_models(self) -> dict:
11
  models = {}
12
  for class_name in [
13
  "sentiment_polarity",
 
1
  import cleantext
2
  import joblib
 
3
  import os
4
 
5
  class PreTrainedPipeline():
6
  def __init__(self, path) -> None:
7
  self.models = self.load_models(path)
8
 
9
+ def load_models(self, path) -> dict:
10
  models = {}
11
  for class_name in [
12
  "sentiment_polarity",