Upload 8 files

analyze_bob_hf.py

@@ -30,11 +30,11 @@ except Exception:
 BASE_DIR = Path(os.environ.get("BOB_BASE_DIR", Path(__file__).parent.parent))
 TRANSCRIPTIONS_DIR = Path(os.environ.get("BOB_TRANSCRIPTIONS_DIR", BASE_DIR / "output" / "transcriptions"))
 OUTPUT_FILE = Path(os.environ.get("BOB_OUTPUT_FILE", BASE_DIR / "output" / "resume_bob.txt"))
-HF_MODEL = os.environ.get("HF_MODEL", "Qwen/Qwen2-7B-Instruct")  # Qwen 7B par défaut
+HF_MODEL = os.environ.get("HF_MODEL", "google/gemma-3-4b-pt")  # Gemma 3 4B par défaut
 
 def get_hf_model():
     """Récupère le modèle Hugging Face depuis la variable d'environnement"""
-    return os.environ.get("HF_MODEL", "Qwen/Qwen2-7B-Instruct")
+    return os.environ.get("HF_MODEL", "google/gemma-3-4b-pt")
 
 
 def load_hf_model():

bob_gui_hf.py

@@ -331,21 +331,16 @@ class BOBProcessorGUI(QMainWindow):
         self.hf_combo = QComboBox()
         self.hf_combo.setFont(QFont("Arial", 11))
         self.hf_combo.addItems([
-            "Qwen/Qwen2-7B-Instruct (recommandé)",
-            "microsoft/DialoGPT-medium (plus léger)",
-            "google/flan-t5-base (très léger)",
+            "google/gemma-3-4b-pt (Gemma 3 4B)",
         ])
-        self.hf_combo.setCurrentText("Qwen/Qwen2-7B-Instruct (recommandé)")
+        self.hf_combo.setCurrentText("google/gemma-3-4b-pt (Gemma 3 4B)")
         self.hf_combo.setStyleSheet("font-size: 11pt; padding: 5px;")
         # Tooltip explicative pour le choix du modèle
         self.hf_combo.setToolTip(
-            "💡 Choix du modèle d'analyse Hugging Face :\n\n"
-            "• Qwen2-7B : Recommandé, excellent pour l'analyse\n"
-            "  Bonne précision, nécessite plus de ressources\n\n"
-            "• DialoGPT-medium : Modèle plus léger\n"
-            "  Utilisez si vous avez peu de RAM\n\n"
-            "• FLAN-T5-base : Très léger\n"
-            "  Pour PC peu puissants ou tests rapides"
+            "💡 Modèle Gemma 3 4B (Google) :\n\n"
+            "• Gemma 3 4B officiel\n"
+            "• Même modèle que la version EXE\n"
+            "• Optimisé pour Hugging Face Spaces"
         )
         config_layout.addWidget(self.hf_combo, 3, 1)
 
@@ -467,20 +462,10 @@ class BOBProcessorGUI(QMainWindow):
         else:
             whisper_model = "small"  # défaut
         
-        # HF selection -> nom de modèle Hugging Face
+        # HF selection -> nom de modèle Hugging Face (Gemma 3 4B uniquement)
         hf_choice = self.hf_combo.currentText()
-        if "Qwen2-7B" in hf_choice:
-            hf_model_name = "Qwen/Qwen2-7B-Instruct"
-            fast_mode = False
-        elif "DialoGPT" in hf_choice:
-            hf_model_name = "microsoft/DialoGPT-medium"
-            fast_mode = True
-        elif "flan-t5" in hf_choice:
-            hf_model_name = "google/flan-t5-base"
-            fast_mode = True
-        else:
-            hf_model_name = "Qwen/Qwen2-7B-Instruct"
-            fast_mode = False
+        hf_model_name = "google/gemma-3-4b-pt"  # Gemma 3 4B
+        fast_mode = True  # Gemma est rapide
 
         # Configuration des variables d'environnement
         os.environ["BOB_INPUT_DIR"] = str(self.input_dir)