Update app.py

app.py

@@ -92,7 +92,7 @@ def sample_tokens(logits, temperature=0.0, top_p=None, top_k=None, margin_confid
 
 # --- Model Loading and Constants ---
 # Load model configuration to get special token IDs
-config = AutoConfig.from_pretrained("Dream-org/Dream-v0-Instruct-7B", trust_remote_code=True)
+config = AutoConfig.from_pretrained("Dream-org/Dream-Coder-v0-Instruct-7B", trust_remote_code=True)
 model_path = "Dream-org/Dream-Coder-v0-Instruct-7B"
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 print(f"Using device: {device}")