SimpleFrog commited on
Commit
15513dc
·
verified ·
1 Parent(s): 3778f8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ st.write("Upload un fichier audio et laisse ton modèle fine-tuné faire le trav
16
  @st.cache_resource # Permet de ne charger qu'une seule fois le modèle
17
  def load_model():
18
  base_model_name = "openai/whisper-large" # Modèle de base
19
- adapter_model_name = "SimpleFrog/whisper_finetuned3" # Adaptateur LoRA
20
 
21
  # Charger le modèle de base
22
  model = WhisperForConditionalGeneration.from_pretrained(base_model_name)
@@ -75,5 +75,5 @@ if uploaded_file is not None:
75
  # Supprimer le fichier temporaire après l'affichage
76
  os.remove(temp_audio_path)
77
 
78
- st.write("🔹 Modèle utilisé :", "Whisper Large + Adaptateur LoRA (SimpleFrog/whisper_finetuned3)")
79
 
 
16
  @st.cache_resource # Permet de ne charger qu'une seule fois le modèle
17
  def load_model():
18
  base_model_name = "openai/whisper-large" # Modèle de base
19
+ adapter_model_name = "SimpleFrog/whisper_finetuned" # Adaptateur LoRA
20
 
21
  # Charger le modèle de base
22
  model = WhisperForConditionalGeneration.from_pretrained(base_model_name)
 
75
  # Supprimer le fichier temporaire après l'affichage
76
  os.remove(temp_audio_path)
77
 
78
+ st.write("🔹 Modèle utilisé :", "Whisper Large + Adaptateur LoRA (SimpleFrog/whisper_finetuned)")
79