Update app.py

app.py

@@ -4,6 +4,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import spacy
 import re
+from typing import List
 
 # Set environment variables for writable directories
 os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
@@ -29,7 +30,7 @@ REPLACE_PRONOUNS = {"he", "she", "they", "He", "She", "They"}
 
 class CorefRequest(BaseModel):
     text: str
-    main_characters: str
+    main_characters: List[str]
 
 def extract_core_name(mention_text, main_characters):
     words = mention_text.split()
@@ -101,7 +102,7 @@ def process_text(text, main_characters):
 
 @app.post("/predict")
 async def predict(coref_request: CorefRequest):
-    resolved_text = process_text(coref_request.text, coref_request.main_characters.split(","))
+    resolved_text = process_text(coref_request.text, coref_request.main_characters)
     if resolved_text:
         return {"resolved_text": resolved_text}
     raise HTTPException(status_code=400, detail="Coreference resolution failed")