Attempt for radio

app.py

@@ -2,6 +2,8 @@ from transformers import pipeline
 import gradio as gr
 from pytube import YouTube
 import os
+import requests
+import time
 
 pipe = pipeline(model="dussen/whisper-small-nl-hc")
 print(pipe)
@@ -48,18 +50,22 @@ def audio_to_text(audio):
     return text
 
 def radio_to_text(radio_url):
-    # A radio livestream
-    stream_url = radio_url
+    stream_url = "https://icecast.omroep.nl/radio1-bb-mp3"
 
     r = requests.get(stream_url, stream=True)
 
+    # Open it and after 10 seconds close the connection
     with open('stream.mp3', 'wb') as f:
+        # Get the stopping time as a UNIX timestamp
+        stop_after = time.time() + 0.5
+
         try:
             for block in r.iter_content(1024):
                 f.write(block)
+                if time.time() > stop_after:
+                    break
         except KeyboardInterrupt:
             pass
-
     text = pipe("stream.mp3")["text"]
     print(text)
     return text