π Bugfix handler for inference endpoints
#4
by
vinri2z
- opened
- handler.py +2 -1
- 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
|