vic3610 commited on
Commit
fd154f2
·
verified ·
1 Parent(s): 765cedd

Upload 8 files

Browse files
Files changed (3) hide show
  1. analyze_bob_hf.py +1 -1
  2. app.py +1 -1
  3. bob_gui_hf.py +7 -7
analyze_bob_hf.py CHANGED
@@ -30,7 +30,7 @@ except Exception:
30
  BASE_DIR = Path(os.environ.get("BOB_BASE_DIR", Path(__file__).parent.parent))
31
  TRANSCRIPTIONS_DIR = Path(os.environ.get("BOB_TRANSCRIPTIONS_DIR", BASE_DIR / "output" / "transcriptions"))
32
  OUTPUT_FILE = Path(os.environ.get("BOB_OUTPUT_FILE", BASE_DIR / "output" / "resume_bob.txt"))
33
- HF_MODEL = os.environ.get("HF_MODEL", "google/gemma-3-4b-pt") # Gemma 3 4B par défaut
34
 
35
  def get_hf_model():
36
  """Récupère le modèle Hugging Face depuis la variable d'environnement"""
 
30
  BASE_DIR = Path(os.environ.get("BOB_BASE_DIR", Path(__file__).parent.parent))
31
  TRANSCRIPTIONS_DIR = Path(os.environ.get("BOB_TRANSCRIPTIONS_DIR", BASE_DIR / "output" / "transcriptions"))
32
  OUTPUT_FILE = Path(os.environ.get("BOB_OUTPUT_FILE", BASE_DIR / "output" / "resume_bob.txt"))
33
+ HF_MODEL = os.environ.get("HF_MODEL", "google/gemma-3-4b-pt") # Gemma 3 4B FORCÉ
34
 
35
  def get_hf_model():
36
  """Récupère le modèle Hugging Face depuis la variable d'environnement"""
app.py CHANGED
@@ -51,7 +51,7 @@ def process_audio_file(audio_file):
51
  os.environ["BOB_TRANSCRIPTIONS_DIR"] = str(transcriptions_dir)
52
  os.environ["BOB_OUTPUT_FILE"] = str(output_dir / "resume_bob.txt")
53
  os.environ["WHISPER_MODEL"] = "small" # Modèle léger pour le Space
54
- os.environ["HF_MODEL"] = "Qwen/Qwen2-7B-Instruct" # Modèle gratuit et léger
55
 
56
  print("🔄 Début de la transcription...")
57
 
 
51
  os.environ["BOB_TRANSCRIPTIONS_DIR"] = str(transcriptions_dir)
52
  os.environ["BOB_OUTPUT_FILE"] = str(output_dir / "resume_bob.txt")
53
  os.environ["WHISPER_MODEL"] = "small" # Modèle léger pour le Space
54
+ os.environ["HF_MODEL"] = "google/gemma-3-4b-pt" # Gemma 3 4B FORCÉ
55
 
56
  print("🔄 Début de la transcription...")
57
 
bob_gui_hf.py CHANGED
@@ -331,15 +331,15 @@ class BOBProcessorGUI(QMainWindow):
331
  self.hf_combo = QComboBox()
332
  self.hf_combo.setFont(QFont("Arial", 11))
333
  self.hf_combo.addItems([
334
- "google/gemma-3-4b-pt (Gemma 3 4B)",
335
  ])
336
- self.hf_combo.setCurrentText("google/gemma-3-4b-pt (Gemma 3 4B)")
337
  self.hf_combo.setStyleSheet("font-size: 11pt; padding: 5px;")
338
  # Tooltip explicative pour le choix du modèle
339
  self.hf_combo.setToolTip(
340
  "💡 Modèle Gemma 3 4B (Google) :\n\n"
341
- "• Gemma 3 4B officiel\n"
342
- "• Même modèle que la version EXE\n"
343
  "• Optimisé pour Hugging Face Spaces"
344
  )
345
  config_layout.addWidget(self.hf_combo, 3, 1)
@@ -462,10 +462,10 @@ class BOBProcessorGUI(QMainWindow):
462
  else:
463
  whisper_model = "small" # défaut
464
 
465
- # HF selection -> nom de modèle Hugging Face (Gemma 3 4B uniquement)
466
  hf_choice = self.hf_combo.currentText()
467
- hf_model_name = "google/gemma-3-4b-pt" # Gemma 3 4B
468
- fast_mode = True # Gemma est rapide
469
 
470
  # Configuration des variables d'environnement
471
  os.environ["BOB_INPUT_DIR"] = str(self.input_dir)
 
331
  self.hf_combo = QComboBox()
332
  self.hf_combo.setFont(QFont("Arial", 11))
333
  self.hf_combo.addItems([
334
+ "google/gemma-3-4b-pt (Gemma 3 4B FORCÉ)",
335
  ])
336
+ self.hf_combo.setCurrentText("google/gemma-3-4b-pt (Gemma 3 4B FORCÉ)")
337
  self.hf_combo.setStyleSheet("font-size: 11pt; padding: 5px;")
338
  # Tooltip explicative pour le choix du modèle
339
  self.hf_combo.setToolTip(
340
  "💡 Modèle Gemma 3 4B (Google) :\n\n"
341
+ "• Gemma 3 4B FORCÉ\n"
342
+ "• Modèle 4B comme demandé\n"
343
  "• Optimisé pour Hugging Face Spaces"
344
  )
345
  config_layout.addWidget(self.hf_combo, 3, 1)
 
462
  else:
463
  whisper_model = "small" # défaut
464
 
465
+ # HF selection -> nom de modèle Hugging Face (Gemma 3 4B FORCÉ)
466
  hf_choice = self.hf_combo.currentText()
467
+ hf_model_name = "google/gemma-3-4b-pt" # Gemma 3 4B FORCÉ
468
+ fast_mode = True # 4B est rapide
469
 
470
  # Configuration des variables d'environnement
471
  os.environ["BOB_INPUT_DIR"] = str(self.input_dir)