Create app.py

app.py

@@ -0,0 +1,21 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline
+
+app = FastAPI()
+
+# Load the model and tokenizer from the Hugging Face Hub
+model_name = "your-username/ner-model"
+nlp = pipeline("ner", model=model_name, tokenizer=model_name)
+
+class TextRequest(BaseModel):
+    text: str
+
+@app.post("/predict/")
+def predict(request: TextRequest):
+    ner_results = nlp(request.text)
+    return ner_results
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)