Upload 3 files

Dockerfile

@@ -1,12 +1,30 @@
-FROM python 3.9
+# Use Official Python image
+FROM python:3.9
+
+# Set the working directory
 WORKDIR /text2text
+
+# Copy the requirements file into the container
 COPY ./requirements.txt /text2text/requirements.txt
+
+# Install the requirements
 RUN pip install --no-cache-dir --upgrade -r /text2text/requirements.txt
-RUN useradd user
+
+# Add a new user
+RUN useradd -m user
+
+# Switch to the new user
 USER user
+
+# Set environment variables
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH 
 
+# Set the working directory for the user
 WORKDIR $HOME/app
+
+# Copy the current directory contents into the container
 COPY --chown=user . $HOME/app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# Define the command to run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,16 +1,16 @@
 from fastapi import FastAPI
 from transformers import pipeline
 
-# Creat a new FastAPI app instance
-app=FastAPI()
+# Create a new FastAPI app instance
+app = FastAPI()
 
 pipe = pipeline("text2text-generation", model="google/flan-t5-small")
 
 @app.get("/")
 def home():
-    return{message: "hello World"}
+    return {"message": "Hello World"}
 
 @app.get("/generate")
-def generate():
-    output=pipe(text)
-    return {"output":output[0]['generated_text']}
+def generate(text: str):
+    output = pipe(text)
+    return {"output": output[0]['generated_text']}