fix path of cache model

Dockerfile

@@ -1,11 +1,18 @@
 # ใช้ Python base image
 FROM python:3.9-slim
 
+# ตั้งค่า cache location ให้ transformers
+ENV HF_HOME=/app/cache
+ENV TRANSFORMERS_CACHE=/app/cache
+
 # ติดตั้ง dependencies
 WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# สร้างโฟลเดอร์ cache และให้สิทธิ์เขียน
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
 # โหลดโมเดลล่วงหน้า
 COPY download_model.py .
 RUN python download_model.py
@@ -16,5 +23,5 @@ COPY . .
 # เปิดพอร์ต 7860 สำหรับ Gradio
 EXPOSE 7860
 
-# คำสั่งรัน FastAPI หรือ Gradio
+# คำสั่งรัน Gradio
 CMD ["python", "interface.py"]

download_model.py

@@ -1,5 +1,10 @@
+import os
 from transformers import pipeline
 
+# กำหนดค่า cache ให้ transformers และ huggingface_hub
+os.environ["HF_HOME"] = "/app/cache"
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+
 print("Downloading pre-trained model...")
 classifier = pipeline("audio-classification", model="AmeerHesham/distilhubert-finetuned-baby_cry")
 print("Model downloaded successfully!")

interface.py

@@ -1,7 +1,12 @@
+import os
 import gradio as gr
 from transformers import pipeline
 import tempfile
 
+# กำหนดค่า cache location ให้ transformers
+os.environ["HF_HOME"] = "/app/cache"
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+
 # โหลดโมเดลล่วงหน้า
 classifier = pipeline("audio-classification", 
                       model="AmeerHesham/distilhubert-finetuned-baby_cry",