Update app.py

app.py

@@ -57,7 +57,7 @@ def create_conversation(sample):
   }
     
 def download_dataset(dataset):
-    dataset = load_dataset("b-mc2/sql-create-context", split="train")
+    dataset = load_dataset(dataset, split="train")
     dataset = dataset.shuffle().select(range(12500))
      
     # Convert dataset to OAI messages