Application files added

.gitignore

@@ -0,0 +1,2 @@
+venv
+venv/

Dockerfile

@@ -0,0 +1,32 @@
+#use the official python 3.9 image 
+FROM python:3.9
+
+#set the working directory to /code 
+WORKDIR /code 
+
+#COPY the current directory contents in the container at /code
+COPY ./requirements.txt /code/requirements.txt
+
+#install the requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set up a new user named "user"
+RUN useradd 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 the user's home directory 
+WORKDIR $HOME/app
+
+#copy the current directory contents into the container at $HOME/app setting the owner to user
+COPY --chown=user . $HOME/app
+
+# start the FASTAPI app on port 7680
+CMD ["uvicorn","app:app","--host", "0.0.0.0","--port","7860"]
+

app.py

@@ -0,0 +1,21 @@
+from fastapi import FastAPI
+from transformers import pipeline
+
+#create a new FASTAPI app instance 
+app = FastAPI()
+
+#initialize the text generation pipeline
+pipe = pipeline("text2text-generation", model = "google/flan-t5-base")
+
+@app.get("/")
+def home():
+    return {"message":"hello world"}
+
+#Define a function to handle the GET request at "/generate"
+@app.get("/generate")
+def generate(text:str):
+    ## use the pipeline to generate text from the given input 
+    output = pipe(text)
+    # return the generated text in the json response 
+    return {"output":output[0]["generated_text"]}
+

requirements.txt

@@ -0,0 +1,6 @@
+fastapi
+requests
+uvicorn[standard]
+sentencepiece
+torch
+transformers