update app.py

app.py

@@ -14,8 +14,8 @@ audio_transcription: str = None
 
 def main ():
     
+    print("Start initiating model")
     init_model()
-    print('Init model successful')
     # x = st.slider('Select a value')
     # st.write(x, 'squared is', x * x)
 
@@ -26,9 +26,10 @@ def main ():
 
     dataset = load_dataset("distil-whisper/librispeech_long", "clean", split="validation")
     sample = dataset[0]["audio"]
-    st.write('Sample:', sample)
+    st.write('Sample:', transcribe(sample))
+
 
-async def init_model ():
+def init_model ():
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
     torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
@@ -53,8 +54,9 @@ async def init_model ():
         torch_dtype=torch_dtype,
         device=device,
     )
+    print('Init model successful')
 
-async def transcribe (audio_sample: bytes, pipe) -> str:
+async def transcribe (audio_sample: bytes) -> str:
 
     # dataset = load_dataset("distil-whisper/librispeech_long", "clean", split="validation")
     # sample = dataset[0]["audio"]