Ajout du modèle de captioning PaliGemma 2

app.py

@@ -1,47 +1,34 @@
 import gradio as gr
 from huggingface_hub import hf_hub_download
 from ctransformers import AutoModelForCausalLM
-import base64
-from PIL import Image
-import io
 
-MODEL_REPO = "TheBloke/Ministral-3-3B-Instruct-2512-GGUF"
-MODEL_FILE = "ministral-3-3b-instruct-2512.Q4_K_M.gguf"
+# French-Alpaca - modéle français confirmé
+MODEL_REPO = "jpacifico/french-alpaca-instruct-Q4-GGUF"
+MODEL_FILE = "french-alpaca-instruct-Q4.gguf"
 
-print("📥 Téléchargement du modèle...")
+print("📥 Téléchargement du modèle français...")
 model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
 
 print("⚙️ Chargement...")
 llm = AutoModelForCausalLM.from_pretrained(
     model_path,
-    model_type="mistral",
+    model_type="mistral",  # French-Alpaca est basé sur Mistral
     threads=2,
     context_length=2048,
     max_new_tokens=256
 )
+print("✅ Modèle français chargé !")
 
 def respond(message, history):
-    prompt = f"<s>[INST] {message} [/INST]"
+    # Format du prompt adapté
+    prompt = f"### Instruction:\n{message}\n\n### Response:\n"
     response = llm(prompt, temperature=0.7, max_new_tokens=256)
     return response.strip()
 
-def image_respond(image, message):
-    # Pour l'analyse d'image (format spécifique)
-    prompt = f"<s>[INST] [IMG] {message} [/INST]"
-    response = llm(prompt, temperature=0.7, max_new_tokens=256)
-    return response.strip()
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 Assistant Mistral Vision (français + images)")
-    
-    with gr.Tab("Chat texte"):
-        chat = gr.ChatInterface(fn=respond)
-    
-    with gr.Tab("Analyse d'image"):
-        image_input = gr.Image(type="pil", label="Votre image")
-        text_input = gr.Textbox(label="Question sur l'image", placeholder="Que voyez-vous sur cette image ?")
-        output = gr.Textbox(label="Réponse")
-        submit = gr.Button("Analyser")
-        submit.click(fn=image_respond, inputs=[image_input, text_input], outputs=output)
+demo = gr.ChatInterface(
+    fn=respond,
+    title="🤖 Assistant Français (French-Alpaca)",
+    description="Modèle entraîné spécifiquement pour le français"
+)
 
 demo.launch(server_name="0.0.0.0", server_port=7860)