Update app.py

app.py

@@ -10,7 +10,12 @@ model_whisper = whisper.load_model("small")
 nlp_ner = spacy.load("es_pipeline")
 
 def main_generator(youtube_id:str):
-  YouTubeID = youtube_id # 
+  YouTubeID = youtube_id.split("https://www.youtube.com/watch?v=") #
+  if len(YouTubeID)>1:
+      YouTubeID = YouTubeID[1]
+  else:
+      YouTubeID ='xOZM-1p-jAk'
+      
   OutputFile = 'test_audio_youtube_.m4a'
 
   os.system(f"youtube-dl -o {OutputFile} {YouTubeID} --extract-audio --restrict-filenames -f 'bestaudio[ext=m4a]'")
@@ -32,9 +37,9 @@ def main_generator(youtube_id:str):
   return {"text": text, "entities": output_list}      
 inputs = [gr.Textbox(lines=1, placeholder="Link of youtube video here...", label="Input")]
 outputs = gr.HighlightedText()
-title="ASR FOR MEDICAL RECORDS"
+title="ASR FOR SPANISH MEDICAL RECORDS"
 description = "This demo uses AI Models to create an AUDIO ANNOTATION FOR MEDICAL RECORDS"
-examples = ['xOZM-1p-jAk']
+examples = ['https://www.youtube.com/watch?v=xOZM-1p-jAk']
 
 io = gr.Interface(fn=main_generator, inputs=inputs, outputs=outputs, title=title, description = description, examples = examples,