fastapi init

.gitignore

@@ -0,0 +1,2 @@
+venv
+__pycache__

main.py

@@ -0,0 +1,23 @@
+from fastapi import FastAPI, Request
+from transformers import pipeline
+from pydantic import BaseModel
+
+class TextQuery(BaseModel):
+    text: str
+
+
+def predict(text: str):
+    model_name = "blanchefort/rubert-base-cased-sentiment-rurewiews"
+    pipe = pipeline("text-classification", model=model_name, return_all_scores=True)
+    scores = pipe(text)[0]
+    sorted_scores = sorted(scores, key=lambda x: x['score'], reverse=True)
+    return sorted_scores
+
+
+app = FastAPI()
+
+@app.post("/", )
+def home(request: Request, q: TextQuery):
+
+
+    return predict(q.text)

requirements.txt

@@ -0,0 +1,58 @@
+absl-py==2.1.0
+annotated-types==0.6.0
+anyio==4.3.0
+astunparse==1.6.3
+cachetools==5.3.3
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+fastapi==0.110.0
+filelock==3.13.1
+flatbuffers==23.5.26
+fsspec==2024.2.0
+gast==0.5.4
+google-auth==2.28.1
+google-auth-oauthlib==1.2.0
+google-pasta==0.2.0
+grpcio==1.62.0
+h11==0.14.0
+h5py==3.10.0
+huggingface-hub==0.21.2
+idna==3.6
+keras==2.15.0
+libclang==16.0.6
+Markdown==3.5.2
+MarkupSafe==2.1.5
+ml-dtypes==0.2.0
+numpy==1.26.4
+oauthlib==3.2.2
+opt-einsum==3.3.0
+packaging==23.2
+protobuf==4.25.3
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pydantic==2.6.3
+pydantic_core==2.16.3
+PyYAML==6.0.1
+regex==2023.12.25
+requests==2.31.0
+requests-oauthlib==1.3.1
+rsa==4.9
+safetensors==0.4.2
+six==1.16.0
+sniffio==1.3.1
+starlette==0.36.3
+tensorboard==2.15.2
+tensorboard-data-server==0.7.2
+tensorflow==2.15.0.post1
+tensorflow-estimator==2.15.0
+tensorflow-io-gcs-filesystem==0.36.0
+termcolor==2.4.0
+tokenizers==0.15.2
+tqdm==4.66.2
+transformers==4.38.1
+typing_extensions==4.10.0
+urllib3==2.2.1
+uvicorn==0.27.1
+Werkzeug==3.0.1
+wrapt==1.14.1