Update app.py

app.py

@@ -1,12 +1,36 @@
+# from fastapi import FastAPI
+# from pydantic import BaseModel
+# from transformers import pipeline
+# import os
+# from huggingface_hub import login
+
+# # Log in to Hugging Face
+# access_token = os.environ.get("ACCESS_TOKEN_1")
+# login(token=access_token, add_to_git_credential=True)
+
+# app = FastAPI()
+
+# # Load the model and tokenizer from the Hugging Face Hub
+# model_name = "MHULO/yembaner"
+# 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=80)
+
+
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import pipeline
 import os
-from huggingface_hub import login
-
-# Log in to Hugging Face
-access_token = os.environ.get("ACCESS_TOKEN_1")
-login(token=access_token, add_to_git_credential=True)
 
 app = FastAPI()
 
@@ -25,3 +49,4 @@ def predict(request: TextRequest):
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=80)
+