Update app.py

app.py

@@ -14,15 +14,14 @@ from dotenv import load_dotenv
 load_dotenv()
 
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-google_genai.configure(api_key=
-GOOGLE_API_KEY)
+google_genai.configure(api_key=GOOGLE_API_KEY)
 
 st.set_page_config(
     page_title="VidText"
 )
 
 st.title("VidText-whisper")
-st.subheading("A video/audio transcription app")
+st.subheader("A video/audio transcription app")
 
 def youtube_video_downloader(url):
     yt_vid = YouTube(url)
@@ -56,7 +55,7 @@ def audio_processing(mp3_audio):
 
 @st.cache_resource
 def load_asr_model():
-    asr_model = pipeline(task="automatic-speech-recognition", model="distil-whisper/distil-large-v3")
+    asr_model = pipeline(task="automatic-speech-recognition", model="openai/whisper-small")
     return asr_model
 
 transcriber_model = load_asr_model()
@@ -83,8 +82,7 @@ with youtube_url_tab:
 
         yt_video, title = youtube_video_downloader(url)
         if url:
-           if st.button("Transcribe", key
-="yturl"):
+           if st.button("Transcribe", key="yturl"):
                with st.spinner("Transcribing..."):
                    with st.spinner('Extracting audio...'):
                        audio = audio_extraction(yt_video)