Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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/
|
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/
|
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 |
|