Fix shi

.env

@@ -1,3 +1 @@
-HF_HOME=/home/container/models
-TRANSFORMERS_CACHE=/home/container/cache
-TMPDIR=/home/container/cache
+HF_HOME=/api/models

Dockerfile

@@ -7,9 +7,6 @@ COPY --chown=user ./ /api/
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 
-RUN mkdir /api/cache
-RUN chmod a+rwx /api/cache
-
 RUN mkdir /api/models
 RUN chmod a+rwx /api/models
 

main.py

@@ -22,7 +22,8 @@ def receive_data():
   print("Prompt:", data["prompt"])
 
   prompt = data["prompt"]
-  
+
+  pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))  
   image = pipe(prompt).images[0]
 
   # Convert the torch Tensor to a NumPy array and move to CPU

requirements.txt

@@ -2,4 +2,5 @@ diffusers
 transformers
 matplotlib
 flask
-torch
+torch
+accelerate