test gemma

Dockerfile

@@ -23,7 +23,8 @@ ENV PATH="/opt/venv/bin:$PATH"
 RUN pip3 install --no-cache-dir -r requirements.txt
 
 # Configuration Ollama
-ENV model="mistral:7b-instruct-q4_K_M"
+#ENV model="mistral:7b-instruct-q4_K_M"
+ENV model="gemma:2b"
 RUN mkdir -p /.ollama && chmod 777 /.ollama
 
 # Exposition des ports (FastAPI et Ollama)

README.md

@@ -1,10 +1,20 @@
 ---
-title: Chat Modele
-emoji: 🔥
-colorFrom: red
-colorTo: gray
+title: Chat Modele API
+emoji: 🤖
+colorFrom: blue
+colorTo: red
 sdk: docker
+app_port: 7860
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Chat Modele API
+
+API FastAPI pour interagir avec le modèle Mistral via Ollama.
+
+## Endpoints
+
+- GET `/` : Message de bienvenue
+- POST `/generate` : Générer du texte
+- GET `/docs` : Documentation Swagger
+- GET `/redoc` : Documentation ReDoc

start.sh

@@ -7,7 +7,7 @@ ollama serve &
 sleep 5
 
 # Télécharger le modèle
-ollama pull mistral:7b-instruct-q4_K_M
+ollama pull gemma:2b
 
 # Activer l'environnement virtuel et démarrer l'API
 source /opt/venv/bin/activate