Update Dockerfile

Dockerfile

@@ -30,9 +30,9 @@ RUN pip install --no-cache-dir \
     sentencepiece \
     accelerate
 
-# Create uploads and cache directories with proper permissions
-RUN mkdir -p /app/uploads /app/cache && \
-    chmod -R 777 /app/uploads /app/cache
+# Create uploads, cache, and fine-tuning directories with proper permissions
+RUN mkdir -p /app/uploads /app/cache /app/pegasus_finetune /app/bert_finetune /app/legalbert_finetune && \
+    chmod -R 777 /app/uploads /app/cache /app/pegasus_finetune /app/bert_finetune /app/legalbert_finetune
 
 # Set environment variable for Hugging Face cache
 ENV HF_HOME=/app/cache