Update app.py

app.py

@@ -49,3 +49,9 @@ st.markdown("Larger models are more accurate, but slower. For 1min video, it'll
 if st.button("Transcribe"):
     transcript = get_transcript(url, model_size, lang, format)
     st.text_area("Transcription of the video", transcript)
+
+    if format == ".srt":
+        srt_file = open("transcription.srt", "w", encoding="utf-8")
+        srt_file.write(transcript)
+        srt_file.close()
+        st.download_button("Download SRT", "transcription.srt")