Simonc-44 commited on
Commit
efbf9a2
·
verified ·
1 Parent(s): 1fecb37

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +27 -0
Dockerfile ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Utilise une image légère avec Python
2
+ FROM python:3.10-slim
3
+
4
+ # Installe les dépendances système
5
+ RUN apt-get update && apt-get install -y git build-essential
6
+
7
+ # Crée un dossier de travail
8
+ WORKDIR /app
9
+
10
+ # Installe les bibliothèques nécessaires
11
+ RUN pip install --no-cache-dir flask transformers accelerate bitsandbytes
12
+
13
+ # Commande pour télécharger ton modèle au démarrage
14
+ # (On utilise la version légère pour que ça tienne sur le CPU gratuit)
15
+ RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; \
16
+ model_id = 'CygnisAI/Cygnis-Alpha-1.7B-v0.1-Instruct'; \
17
+ AutoTokenizer.from_pretrained(model_id); \
18
+ AutoModelForCausalLM.from_pretrained(model_id)"
19
+
20
+ # Copie ton script serveur (on va le créer juste après)
21
+ COPY app.py .
22
+
23
+ # Expose le port pour Hugging Face
24
+ EXPOSE 7860
25
+
26
+ # Lance le serveur
27
+ CMD ["python", "app.py"]