upload project

Dockerfile

@@ -0,0 +1,7 @@
+FROM python:3.9
+
+COPY . .
+
+RUN pip install --upgrade -r requirements.txt
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,38 @@
+import logging
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline
+
+logging.basicConfig(
+    format='%(asctime)s.%(msecs)03d %(levelname)-8s %(message)s',
+    level=logging.DEBUG,
+    datefmt='%Y-%m-%d %H:%M:%S'
+)
+classifier = pipeline("zero-shot-classification", model="models/classificator", use_fast=True)
+app = FastAPI()
+
+
+class RequestData(BaseModel):
+    multiLabel: bool
+    sequence: str
+    labels: list[str]
+
+
+class ResponseData(BaseModel):
+    sequence: str
+    labels: list[str]
+    scores: list[float]
+
+
+@app.post("/classify", response_model=ResponseData, tags=["Classificator"])
+async def classify_text(data: RequestData):
+    result = classifier(data.sequence, data.labels, multi_label=data.multiLabel)
+    logging.info(result)
+
+    return result
+
+
+@app.get("/ping", tags=["TEST"])
+def ping():
+    return "pong"
+