added files

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.11.2-slim
+
+EXPOSE 8000
+WORKDIR /app
+
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends netcat && \
+    apt-get -y install libpq-dev gcc && \
+    rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+COPY requirements.txt ./
+RUN pip install -r requirements.txt
+
+ADD . /app
+
+ENTRYPOINT ./entrypoint.sh

docker-compose.yml

@@ -0,0 +1,18 @@
+version: '3'
+
+services:
+  news_classification:
+    container_name: news_classification
+    image: news_classification:latest
+    build:
+      context: .
+      dockerfile: Dockerfile
+    restart: always
+    ports:
+      - "8000:8000"
+    environment:
+      IMAGE: "news_classification"
+    logging:
+      options:
+        max-file: "5"
+        max-size: "10m"

entrypoint.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+uvicorn main:app --host=0.0.0.0 --reload --port 8000

main.py

@@ -0,0 +1,14 @@
+from transformers import pipeline
+from fastapi import FastAPI
+
+app = FastAPI()
+
+classifier = pipeline("zero-shot-classification", model='cross-encoder/nli-deberta-v3-base')
+
+
+@app.get("/")
+async def root(title):
+    categories = ["technology", "sports", "politics", "weather", "business", "entertainment"]
+    response = classifier(title, categories)
+    response["category"] = response["labels"][0]
+    return response

requirements.txt

@@ -0,0 +1,34 @@
+anyio==3.6.2
+certifi==2023.5.7
+charset-normalizer==3.1.0
+click==8.1.3
+fastapi==0.95.2
+filelock==3.12.0
+fsspec==2023.5.0
+h11==0.14.0
+huggingface-hub==0.14.1
+idna==3.4
+Jinja2==3.1.2
+MarkupSafe==2.1.2
+mpmath==1.3.0
+networkx==3.1
+numpy==1.24.3
+packaging==23.1
+Pillow==9.5.0
+protobuf==3.20.0
+pydantic==1.10.8
+PyYAML==6.0
+regex==2023.5.5
+requests==2.31.0
+sentencepiece==0.1.99
+sniffio==1.3.0
+starlette==0.27.0
+sympy==1.12
+tokenizers==0.13.3
+torch==2.0.1
+torchvision==0.15.2
+tqdm==4.65.0
+transformers==4.29.2
+typing_extensions==4.6.2
+urllib3==2.0.2
+uvicorn==0.22.0