Upload 8 files
Browse files- analyze_bob_hf.py +1 -1
- app.py +1 -1
- 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
|
| 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"] = "
|
| 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
|
| 342 |
-
"•
|
| 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
|
| 466 |
hf_choice = self.hf_combo.currentText()
|
| 467 |
-
hf_model_name = "google/gemma-3-4b-pt" # Gemma 3 4B
|
| 468 |
-
fast_mode = True #
|
| 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)
|