Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import pipeline
 from pydub import AudioSegment
 
 pipe = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
-
+#morse code dictionary
 dictionary = {' ': '/', 
 	"'": '.----.', 
 	'0': '-----', 
@@ -48,16 +48,16 @@ dictionary = {' ': '/',
 
 
 def asr(speech):
-  dummy = AudioSegment.from_file("dummy.mp3", format="mp3")
+  dummy = AudioSegment.from_file("dummy.mp3", format="mp3") #load audio files
   dot = AudioSegment.from_file("dot.mp3", format="mp3")
   dash = AudioSegment.from_file("dash.mp3", format="mp3")
   space = AudioSegment.from_file("space.mp3", format="mp3")
   i=0
-  transcript = pipe(speech)['text']
+  transcript = pipe(speech)['text'] #convert input speech to text
   morseCode = ""
   for character in transcript:
     morseCode += dictionary[character] + " "
-  while i<len(morseCode):
+  while i<len(morseCode): #append dash and dot sounds
     if morseCode[i] == ".":
       dummy = dummy + dot
       i = i+1
@@ -69,7 +69,7 @@ def asr(speech):
       i = i+1
     else:
       i = i+1
-  morseAudio = dummy.export("output.mp3", format="mp3")
+  morseAudio = dummy.export("output.mp3", format="mp3") #export final audio
   return transcript, morseCode, "output.mp3"
   
 gr.Interface(fn=asr,
@@ -79,7 +79,7 @@ gr.Interface(fn=asr,
 						gr.outputs.Textbox(type="str", label="Morse Code"),
 						gr.outputs.Audio(type="file", label="Audio Translation")],
              title = "Speech to Morse Code",
-             description = "This app will translate your speech into text, morse code, and audio.",
-			 examples = ["helloworld.mp3"],
+             description = "This app will translate your speech into text, morse code, and audio using wav2vec2-base-960h",
+			 examples = [["helloworld.mp3"], ["SOS.mp3"], ["SMS.mp3]"]],
 			 article = "Author: <a href=\"https://huggingface.co/epdavid2\">Edson</a>"
              ).launch()