Update app.py

app.py

@@ -108,7 +108,6 @@ examples = [
 
 
 def ner(text: str, labels: str, threshold: float, nested_ner: bool) -> Dict[str, Union[str, int, float]]:
-    labels = labels.split(",")
     return [
         {
             "entity": entity["label"],
@@ -117,7 +116,7 @@ def ner(text: str, labels: str, threshold: float, nested_ner: bool) -> Dict[str,
             "end": entity["end"],
         }
         for entity in model.predict_entities(
-            text, labels, flat_ner=not nested_ner, threshold=threshold
+            text, [label.strip() for label in labels.split(",")], flat_ner=not nested_ner, threshold=threshold
         )
     ]
 
@@ -154,9 +153,9 @@ demo = gr.Interface(
 
 app = FastAPI()
 
-@app.get("/my_predict")
-def read_main(text: str = ""):
-    predict_response = requests.post('http://localhost:7860/call/predict', json={'data': [text, 'pokemon', 0.3, False]}).json()
+@app.get("/predict")
+def read_main(text: str = "", labels: str = "", threshold: float = 0.3, nested_ner: bool = False):
+    predict_response = requests.post('http://localhost:7860/call/predict', json={'data': [text, labels, threshold, nested_ner]}).json()
     if "event_id" not in predict_response:
         return predict_response
     return json.loads(requests.get(f'http://localhost:7860/call/predict/{predict_response["event_id"]}').text.split("data: ", 1)[-1])