Updated Dockerfile

Dockerfile

@@ -8,4 +8,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

docker-compose.yaml

@@ -0,0 +1,11 @@
+version: "3"
+
+services:
+  taghub:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    image: kazoo/taghub:local
+    container_name: taghub
+    ports:
+      - 7860:7860

main.py

@@ -1,7 +1,11 @@
+from transformers import pipeline
 from fastapi import FastAPI
 
 app = FastAPI()
 
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 
-def read_root():
-    return {"Hello": "World!"}
+@app.get("/")
+def t5(input):
+    output = pipe_flan(input)
+    return {"output": output[0]["generated_text"]}

req.txt

@@ -1,7 +0,0 @@
-fastapi==0.101.0
-uvicorn==0.23.2
-transformers[torch]==4.31.0
-pillow==10.0.0
-requests==2.31.0
-pydantic==2.1.1
-python-multipart==0.0.6

requirements.txt

@@ -0,0 +1,6 @@
+fastapi==0.74.*
+requests==2.27.*
+sentencepiece==0.1.*
+torch==1.11.*
+transformers==4.*
+uvicorn[standard]==0.17.*