Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 from custom_model import CustomModel
 
 # Load the model and tokenizer
-model_name = "deepseek-ai/DeepSeek-V3"
+model_name = "deepseek-ai/DeepSeek-V3@main"  # Pin a specific revision, e.g., "main"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)
 model = CustomModel.from_pretrained(model_name, config=config, trust_remote_code=True)