Update app.py

app.py

@@ -1,5 +1,6 @@
 
 
+
 # import module
 import streamlit as st
 
@@ -12,11 +13,17 @@ dataset=""
 split=""
 skip=0
 
-column_with_audio="audio_transcription"
-column_with_english_text="en"
-column_with_other_text="transcription"
 
 def load():
+    if dataset=="nlewins/onetalk_questions_full_audio":
+        column_with_audio="audio_transcription"
+        column_with_english_text="en"
+        column_with_other_text="transcription"
+    elif dataset=="nlewins/LSK_full_with_audio":
+        column_with_audio="audio_transcription"
+        column_with_english_text="en"
+        column_with_other_text="transcription"
+    
     ds = datasets.load_dataset(dataset, token=access_token, split=datasets.ReadInstruction("test",from_=skip,to=skip+50))
     for example in ds:
         df=pd.DataFrame([example[column_with_other_text],example[column_with_english_text]])