refactor

Dockerfile

@@ -6,6 +6,10 @@ RUN useradd -m -u 1000 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/user/app
 
@@ -14,8 +18,6 @@ COPY --chown=user . /home/user/app
 
 RUN pip install -r requirements.txt
 
-ENV HF_TOKEN $HF_TOKEN
-
 EXPOSE 7860
 
 CMD ["python", "app.py"]

app.py

@@ -10,6 +10,8 @@ logging.info(os.environ["HF_TOKEN"][:5])
 
 pipe_flan = transformers.pipeline("text2text-generation", model="google/flan-t5-small")
 def google_flan(input_text):
+    print("New response")
+    logging.info("New response")
     return pipe_flan(input_text)
 
 demo = gr.Interface(fn=google_flan, inputs="text", outputs="text")