Upload 4 files

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+COPY requirements.txt /app/
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app.py /app/
+COPY start.sh /app/
+RUN chmod +x /app/start.sh
+
+EXPOSE 7860
+CMD ["/app/start.sh"]

app.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline
+
+app = FastAPI()
+generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
+
+class InputText(BaseModel):
+    prompt: str
+
+@app.get("/")
+def home():
+    return {"message": "Welcome to my Hugging Face Docker App!"}
+
+@app.post("/predict")
+def predict(data: InputText):
+    result = generator(data.prompt, max_length=40)
+    return {"result": result[0]["generated_text"]}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn[standard]
+transformers
+torch

start.sh

@@ -0,0 +1,2 @@
+#!/usr/bin/env bash
+uvicorn app:app --host 0.0.0.0 --port 7860