πŸ› Bugfix handler for inference endpoints

#4
by vinri2z - opened
Files changed (2) hide show
  1. handler.py +2 -1
  2. requirements.txt +2 -1
handler.py CHANGED
@@ -30,4 +30,5 @@ class EndpointHandler:
30
  "end": entity["char_end_index"],
31
  "score": entity["score"],
32
  }
33
- for entity in self.model.predict(data["inputs"])
 
 
30
  "end": entity["char_end_index"],
31
  "score": entity["score"],
32
  }
33
+ for entity in self.model.predict(data["inputs"])
34
+ ]
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- span_marker
 
 
1
+ span_marker
2
+ huggingface_hub==0.23.5