Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
 import pickle
+import whisper
 import streamlit as st
 import torchaudio as ta
 
@@ -70,15 +71,15 @@ if submit_button and uploaded_files is not None:
 
         input_features = processor(resampled_inp[0], sampling_rate=16000, return_tensors='pt').input_features
 
-        
-        lang = detect_language(input_features)
-        
-        with open('languages.pkl', 'rb') as f:
-            lang_dict = pickle.load(f)
-        detected_language = lang_dict[lang]
-
         if task == "translate":
-            ## Here Generate specific language!!!
+            
+            # Detect Language 
+            lang = detect_language(input_features)
+            with open('languages.pkl', 'rb') as f:
+                lang_dict = pickle.load(f)
+            detected_language = lang_dict[lang]
+
+            # Set decoder & Predict translation
             forced_decoder_ids = processor.get_decoder_prompt_ids(language=detected_language, task="translate")
             predicted_ids = model.generate(input_features, forced_decoder_ids=forced_decoder_ids)
         else: