loadenv

Signed-off-by: Rahul D Shetty <35rahuldshetty@gmail.com>

.gitignore

@@ -0,0 +1 @@
+.env

create_synthetic_dataset.py

@@ -10,7 +10,8 @@ from datasets import Dataset
 from miditok import REMI, TokenizerConfig
 from miditoolkit import MidiFile, Instrument, Note, TempoChange
 from tqdm import tqdm
-
+from dotenv import load_dotenv
+load_dotenv()
 
 def create_simple_midi(genre, mood, tempo, num_notes=50):
     """Create a simple synthetic MIDI file."""

inference_midi_gpt.py

@@ -21,7 +21,8 @@ import torch
 from miditok import REMI
 from miditoolkit import MidiFile
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
-
+from dotenv import load_dotenv
+load_dotenv()
 
 def compute_special_token_offset(text_vocab_size: int):
     return {

inference_midi_qwen3.py

@@ -21,6 +21,8 @@ import torch
 from miditok import REMI
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
+from dotenv import load_dotenv
+load_dotenv()
 
 def generate_midi(
     model,

prepare_dataset.py

@@ -19,6 +19,9 @@ from miditoolkit import MidiFile
 from tqdm import tqdm
 from transformers import GPT2Tokenizer
 
+from dotenv import load_dotenv
+load_dotenv()
+
 
 def build_prompt(example):
     """Build a text prompt from dataset metadata fields."""

prepare_dataset_qwen3.py

@@ -19,6 +19,9 @@ from tqdm import tqdm
 from transformers import AutoTokenizer
 
 
+from dotenv import load_dotenv
+load_dotenv()
+
 def build_rich_prompt(example):
     """Build a comprehensive text prompt from all available music metadata."""
     parts = []

test_end_to_end.py

@@ -9,6 +9,9 @@ from miditok import REMI, TokenizerConfig
 from transformers import GPT2Tokenizer
 import io
 
+from dotenv import load_dotenv
+load_dotenv()
+
 # Add app to path
 sys.path.insert(0, "/app")
 

test_qwen3_e2e.py

@@ -9,6 +9,9 @@ from datasets import load_from_disk
 from miditok import REMI
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
+from dotenv import load_dotenv
+load_dotenv()
+
 sys.path.insert(0, "/app")
 
 print("=" * 70)

train_midi_gpt.py

@@ -18,6 +18,9 @@ import os
 import random
 from pathlib import Path
 
+from dotenv import load_dotenv
+load_dotenv()
+
 import numpy as np
 import torch
 from datasets import load_from_disk

train_midi_qwen3.py

@@ -31,6 +31,9 @@ from transformers import (
     TrainerCallback,
 )
 
+from dotenv import load_dotenv
+load_dotenv()
+
 
 # Special token strings that will be added to the tokenizer
 BOS_MIDI_TOKEN = "<|midi_start|>"