Spaces:

TypicaAI
/

magbert-ner-api

Sleeping

hassoudi commited on Apr 25

Commit

6a362a8

verified ·

1 Parent(s): 5c6714e

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, Depends, HTTPException, Request
 from fastapi.security import APIKeyHeader
 from pydantic import BaseModel
 from model import ner_pipeline
@@ -100,15 +101,15 @@ def list_users(x_api_key: str = Depends(token_header)):
         raise HTTPException(status_code=403, detail="Admin access required")
     return {"users": API_KEYS}
 @app.get("/", include_in_schema=False)
 def root():
     return HTMLResponse("""
     <html>
-      <head><title>MagBERT-NER API</title></head>
-      <body style="font-family:sans-serif;">
-        <h1>✅ MagBERT-NER API is running!</h1>
-        <p>This is a private FastAPI app.</p>
-        <p>Visit <a href="/docs">/docs</a> to explore the API interface.</p>
       </body>
     </html>
     """)

 from fastapi import FastAPI, Depends, HTTPException, Request
+from fastapi.responses import HTMLResponse  # ✅ Add this line
 from fastapi.security import APIKeyHeader
 from pydantic import BaseModel
 from model import ner_pipeline
         raise HTTPException(status_code=403, detail="Admin access required")
     return {"users": API_KEYS}
 @app.get("/", include_in_schema=False)
 def root():
     return HTMLResponse("""
     <html>
+      <head><title>NER API</title></head>
+      <body>
+        <h1>✅ NER API is running!</h1>
+        <p>Visit <a href='/docs'>/docs</a> to try the API.</p>
       </body>
     </html>
     """)