fix runtime error

Dockerfile

@@ -14,6 +14,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy source code
 COPY app.py .
 RUN mkdir -p /app/cache /app/flagged
+RUN mkdir -p /tmp/hf_cache /tmp/hf_home
+
 
 EXPOSE 7860
 

app.py

@@ -1,3 +1,7 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/tmp/hf_home"
+
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 import gradio as gr