Update Dockerfile

Dockerfile

@@ -1,31 +1,25 @@
-# Gunakan image Python resmi
-FROM python:3.9-slim@sha256:bef8d69306a7905f55cd523f5604de1dde45bbf745ba896dbb89f6d15c727170
+# Use a lightweight Python base image
+FROM python:3.10-slim
 
 # Set working directory
 WORKDIR /app
 
-# Buat direktori cache dan atur izin
-RUN mkdir -p /app/cache && chmod -R 777 /app/cache
-
-# Bersihkan cache lama (untuk mencegah error izin)
-RUN rm -rf /app/cache/* && chmod -R 777 /app/cache
-
-# Upgrade pip untuk kompatibilitas paket terbaru
-RUN pip install --upgrade pip
-
-# Salin requirements.txt dan install dependensi
+# Copy requirements.txt
 COPY requirements.txt .
+
+# Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Salin kode aplikasi
-COPY app.py .
+# Copy application code
+COPY . .
 
-# Atur environment variable untuk cache
+# Set environment variables for Hugging Face cache
 ENV HF_HOME=/app/cache
 ENV TRANSFORMERS_CACHE=/app/cache
+ENV USE_QUANTIZATION=0  # Set to 1 if using quantization and GPU is available
 
-# Pastikan izin direktori cache tetap benar sebelum menjalankan aplikasi
-RUN chmod -R 777 /app/cache
+# Expose port for FastAPI
+EXPOSE 8000
 
-# Jalankan aplikasi dengan uvicorn
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run the FastAPI app with uvicorn
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]