Update Dockerfile

Dockerfile

@@ -1,31 +1,28 @@
-# Use a more lightweight base image
-FROM python:3.9-slim
+from fastapi import FastAPI
+from transformers import pipeline
 
-# Install necessary system dependencies
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    && rm -rf /var/lib/apt/lists/*
+## create a new FASTAPI app instance
+app=FastAPI()
 
-# Create a new user
-RUN useradd -ms /bin/bash myuser
+# Initialize the text generation pipeline
+pipe = pipeline("text2text-generation", model="google/flan-t5-small")
 
-# Set the working directory in the container to the home directory of the new user
-WORKDIR /home/myuser
 
-# Copy the current directory contents into the container at the home directory
-COPY . .
+@app.get("/")
+def home():
+    return {"message":"Hello World"}
 
-# Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+# Define a function to handle the GET request at `/generate`
 
-# Make port 5000 available to the world outside this container
-EXPOSE 5000
 
-# Define environment variable
-ENV NAME World
+@app.get("/generate")
+def generate(text:str):
+    ## use the pipeline to generate text from given input text
+    output=pipe(text)
+
+    ## return the generate text in Json reposne
+    return {"output":output[0]['generated_text']}
 
-# Run app.py when the container launches
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "5000"]