Update server.py

server.py

@@ -11,12 +11,12 @@ class ToxicAPI(ls.LitAPI):
 
         tokenizer = AutoTokenizer.from_pretrained(
             model_name,
-            use_auth_token=os.getenv("HF_TOKEN"),
+            token=os.getenv("HF_TOKEN"),
             trust_remote_code=True,
         )
         model = AutoModelForSequenceClassification.from_pretrained(
             model_name,
-            use_auth_token=os.getenv("HF_TOKEN"),
+            token=os.getenv("HF_TOKEN"),
             trust_remote_code=True,
         )
 
@@ -33,7 +33,7 @@ class ToxicAPI(ls.LitAPI):
         return self.pipeline(query)
 
     def encode_response(self, output):
-        return output  # already a JSON-serializable list of dicts
+        return output 
 
     def authorize(self, auth: HTTPAuthorizationCredentials = Depends(HTTPBearer())):
         if auth.scheme != "Bearer" or auth.credentials != os.getenv("auth_token"):