fix runtime error

Dockerfile

@@ -8,4 +8,10 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 COPY . .
 
+# Tạo thư mục cache và cấp quyền ghi
+RUN mkdir -p /app/cache
+
+# Đặt biến môi trường để Transformers ghi cache ở nơi được phép
+ENV TRANSFORMERS_CACHE=/app/cache
+
 CMD ["python", "app.py"]

app.py

@@ -1,7 +1,11 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+
 import gradio as gr
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 
 model_id = "NlpHUST/gpt2-vietnamese"
+
 tokenizer = GPT2Tokenizer.from_pretrained(model_id)
 model = GPT2LMHeadModel.from_pretrained(model_id)
 
@@ -16,9 +20,16 @@ def generate(prompt, max_length=100, temperature=0.8):
     )
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
-demo = gr.Interface(fn=generate,
-                    inputs=["text", gr.Slider(50, 200, value=100), gr.Slider(0.1, 1.0, value=0.8)],
-                    outputs="text",
-                    title="Sinh văn bản tiếng Việt bằng GPT-2")
+demo = gr.Interface(
+    fn=generate,
+    inputs=[
+        gr.Textbox(label="Nhập nội dung đầu vào"),
+        gr.Slider(50, 200, value=100, step=10, label="Độ dài tối đa"),
+        gr.Slider(0.1, 1.0, value=0.8, step=0.05, label="Temperature")
+    ],
+    outputs="text",
+    title="Sinh văn bản tiếng Việt bằng GPT-2",
+    description="Dùng mô hình GPT-2 Vietnamese từ NlpHUST để sinh văn bản tiếng Việt."
+)
 
 demo.launch()