Update app.py

app.py

@@ -6,6 +6,7 @@ import whisper
 
 os.system('pip install https://huggingface.co/Armandoliv/es_pipeline/resolve/main/es_pipeline-any-py3-none-any.whl')
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+model_whisper = whisper.load_model("small")
 nlp_ner = spacy.load("es_pipeline")
 
 def main_generator(youtube_id:str):
@@ -13,9 +14,11 @@ def main_generator(youtube_id:str):
   OutputFile = 'test_audio_youtube_.m4a'
 
   os.system(f"youtube-dl -o {OutputFile} {YouTubeID} --extract-audio --restrict-filenames -f 'bestaudio[ext=m4a]'")
-  model_whisper = whisper.load_model("small")
+  
   result = model_whisper.transcribe(OutputFile)
   text = result['text']
+  doc = nlp_ner(text)
+
   output_list = []
   for ent in doc.ents:
       result_dict = {