Hopefully fixed shi

Dockerfile

@@ -2,11 +2,13 @@ FROM python:3.11
 
 WORKDIR /api
 
-COPY ./ /api/
-
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
 RUN mkdir /api/cache
 RUN chmod a+rwx /api/cache
 
+COPY ./ /api/
+
 EXPOSE 7860
 
 CMD ["bash", "start.sh"]

start.sh

@@ -1,4 +1 @@
-pip install --upgrade pip
-pip install -r requirements.txt
-
 python tld/gen_img.py

tld/gen_img.py

@@ -1,12 +1,9 @@
 import io
 import asyncio
 import os
-from PIL import Image
-import numpy as np
 
 import torch
 import torchvision.transforms as transforms
-from torchvision import utils as vutils
 
 from diffusion import DiffusionTransformer, LTDConfig