Update Dockerfile and app.py, add

Dockerfile

@@ -22,7 +22,10 @@ ENV PATH=/home/user/.local/bin:$PATH
 WORKDIR $HOME/app
  
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
+
 COPY --chown=user . $HOME/app
+
  
 # Start the FastAPI app on port 7860, the default port expected by Spaces
+EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--root-path", "/docs"]

app.py

@@ -2,7 +2,11 @@ from fastapi import FastAPI
 from transformers import pipeline
  
 # Create a new FastAPI app instance
-app = FastAPI()
+app = FastAPI(
+    title="demo",
+    version=0.1,
+    root_path="/docs/"
+)
  
 # Initialize the text generation pipeline
 # This function will be able to generate text
@@ -10,12 +14,16 @@ app = FastAPI()
 pipe = pipeline("text2text-generation", 
 model="google/flan-t5-small")
 
+@app.get("/")
+async def root():
+    return {"message": "Hello World"}
+
 # Define a function to handle the GET request at `/generate`
 # The generate() function is defined as a FastAPI route that takes a 
 # string parameter called text. The function generates text based on the # input using the pipeline() object, and returns a JSON response 
 # containing the generated text under the key "output"
 @app.get("/generate")
-def generate(text: str):
+async def generate(text: str):
     """
     Using the text2text-generation pipeline from `transformers`, generate text
     from the given input text. The model used is `google/flan-t5-small`, which

docker-compose.yml

@@ -0,0 +1,24 @@
+version: '3.8'  # Define the version of the docker-compose
+
+services:
+
+  web:  # Name of the service
+    # env_file: .env
+    # image: python-env-web
+    build:
+      context: .  # Context is the current directory
+      dockerfile: Dockerfile  # Name of the Dockerfile
+      # args:
+        # PYTHON_IMG: 3.9-alpine  # Use Python 3.9 as base image
+    ports:
+      - "7860:7860"  # Map host port to container port
+    volumes:
+      - .:/home/app  # Bind mount the current directory to the container directory
+      # - ./config:/home/nonroot/.config  # Bind mount the current directory to the container directory
+    # networks:
+    #   - main
+    stdin_open: true 
+    tty: true
+
+# networks:
+#   main:  # Define a custom network