Update README.md

README.md

@@ -81,7 +81,7 @@ from premsql.agents.tools import SimpleMatplotlibTool
 from premsql.executors import SQLiteExecutor
 
 text2_sqlmodel = Text2SQLGeneratorHF(
-    model_or_name_or_path="premai-io/prem-1B-SQL",
+    model_or_name_or_path="prem-research/prem-1B-SQL",
     experiment_name="test_generators",
     device="cuda:0",
     type="test"
@@ -128,7 +128,7 @@ dataset = bird_dataset = Text2SQLDataset(
 
 # Define a generator
 generator = Text2SQLGeneratorHF(
-    model_or_name_or_path="premai-io/prem-1B-SQL",
+    model_or_name_or_path="prem-research/prem-1B-SQL",
     experiment_name="test_generators",
     device="cuda:0",
     type="test"
@@ -172,9 +172,9 @@ we provide.
 
 Prem-1B-SQL is trained using the following datasets:
 
-1. [BirdBench Training dataset](https://bird-bench.github.io/) | Uploaded on [PremSQL datasets on HF](https://huggingface.co/datasets/premai-io/birdbench)
-2. [Spider dataset](https://yale-lily.github.io/spider) | Uploaded on [PremSQL datasets on HF](https://huggingface.co/datasets/premai-io/spider)
-3. [Domain specialization dataset, gathered and uploaded to PremSQL datasets](https://huggingface.co/datasets/premai-io/domains)
+1. [BirdBench Training dataset](https://bird-bench.github.io/) | Uploaded on [PremSQL datasets on HF](https://huggingface.co/datasets/prem-research/birdbench)
+2. [Spider dataset](https://yale-lily.github.io/spider) | Uploaded on [PremSQL datasets on HF](https://huggingface.co/datasets/prem-research/spider)
+3. [Domain specialization dataset, gathered and uploaded to PremSQL datasets](https://huggingface.co/datasets/prem-research/domains)
 4. [Gretel AI synthetic dataset](https://huggingface.co/datasets/gretelai/synthetic_text_to_sql?row=0)
 
 Additionally we made error handling datasets on top of these datasets to make the model learn from its errors and self correct them.