Add application file

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+EXPOSE 7860
+
+CMD ["python", "main.py"]

main.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI, HTTPException, Depends, Request
+import models
+from transformers import pipeline
+import uvicorn
+
+
+app = FastAPI()
+
+@app.get('/detect_lang')
+async def detect_lang(text: models.Text):
+    try:
+        pipe(text.text)
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f'{e}')
+
+if __name__ == "__main__":
+    pipe = pipeline("text-classification", model="rifkat/uz_kr_lang-detection")
+    uvicorn.run(app, host="0.0.0.0", port=7860)

models.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+
+class Text(BaseModel):
+    text: str 

requirements.txt

@@ -0,0 +1,4 @@
+fastapi==0.81.0
+uvicorn==0.18.3
+pydantic==1.9.2
+transformers