Update Dockerfile

Dockerfile

@@ -1,16 +1,19 @@
-FROM ghcr.io/berriai/litellm:latest
+FROM python:3.11-slim
+
+# Installer les dépendances
+RUN pip install litellm fastapi uvicorn
 
-# Définir un répertoire de travail
 WORKDIR /app
 
-# Copier le fichier de configuration
+# Copier les fichiers nécessaires
+COPY app.py .
 COPY config.yaml .
 
-# Utiliser SQLite comme base de données
-ENV DATABASE_URL=file:/tmp/litellm.db
+# Variables d'environnement
+ENV GEMINI_API_KEY=${GEMINI_API_KEY}
 
 # Exposer le port
 EXPOSE 7860
 
-# Commande par défaut
-CMD ["--port", "7860", "--config", "config.yaml"]
+# Commande de démarrage
+CMD ["uvicorn", "app:predict", "--host", "0.0.0.0", "--port", "7860"]