nimraaslam7777 commited on
Commit
6eb5b0e
·
verified ·
1 Parent(s): 331aeca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -15
app.py CHANGED
@@ -1,23 +1,31 @@
1
- # @title Default title text
2
- from fastapi import FastAPI
3
- from transformers import pipeline
4
 
5
- #create nea fastAPI app instance
6
- app=FastAPI()
7
 
8
- # Use a pipeline as a high-level helper
 
9
 
 
 
10
 
 
 
 
 
11
 
12
- pipeline=pipeline("text2text-generation", model="google/flan-t5-small")
13
 
14
- @app.get("/")
15
- def home():
16
- return {"messag":" hellow word"}
17
- @app.get("/generate")
18
- def generate(text:str):
19
- # Initialize the text generation pipeline
20
 
21
- output=pipe(text)
 
 
 
 
 
 
 
22
 
23
- return {"output_text":output}[0]['generated_text']
 
1
+ ## Use the official Python 3.9 image
2
+ FROM python:3.9
 
3
 
4
+ ## set the working directory to /code
5
+ WORKDIR /code
6
 
7
+ ## Copy the current directory contents in the container at /code
8
+ COPY ./requirements.txt /code/requirements.txt
9
 
10
+ ## Install the requirements.txt
11
+ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
12
 
13
+ # Set up a new user named "user"
14
+ RUN useradd user
15
+ # Switch to the "user" user
16
+ USER user
17
 
18
+ # Set home to the user's home directory
19
 
20
+ ENV HOME=/home/user \
21
+ PATH=/home/user/.local/bin:$PATH
 
 
 
 
22
 
23
+ # Set the working directory to the user's home directory
24
+ WORKDIR $HOME/app
25
+
26
+ # Copy the current directory contents into the container at $HOME/app setting the owner to the user
27
+ COPY --chown=user . $HOME/app
28
+
29
+ ## Start the FASTAPI App on port 7860
30
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
31