setup done

Dockerfile

@@ -0,0 +1,29 @@
+# Use an official Python runtime as a parent image
+FROM python:3.9
+
+# Set the working directory to /code in the container
+WORKDIR /code
+
+# Copy requirements.txt into the container at /code
+COPY ./requirements.txt /code/requirements.txt
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \\
+    PATH=/home/user/.local/bin:$PATH
+
+# Set the working directory to $HOME/app in the container
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app
+COPY --chown=user . $HOME/app
+
+# Make port 7860 available to the world outside this container
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+

app.py

@@ -0,0 +1,24 @@
+from fastapi import FastAPI
+from transformers import pipeline
+
+# Create a FastAPI instance
+app = FastAPI()
+
+# The root endpoint of the application
+@app.get("/")
+def root():
+    return {'message': 'Welcome to the Text Generation API!'}
+
+# Initialize the text generation pipeline
+pipe = pipeline('text2test-generation', model='google/flan-t5-small')
+
+# The GET endpoint of the application,
+# which corresponds to the text generation
+# functionality. The generate() method takes 
+# a prompt as input and returns the generated
+# text as output in the form of a JSON object.
+@app.get("/generate")
+def generate(prompt: str):
+    output = pipe(prompt)
+    return {'output': output[0]['generated_text']}
+

requirements.txt

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