diff --git a/README.md b/README.md
index f8df02c..e96f1f5 100644
--- a/README.md
+++ b/README.md
@@ -18,7 +18,7 @@
   <a href="https://huggingface.co/deepseek-ai" target="_blank">
     <img alt="Hugging Face" src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-DeepSeek%20AI-ffc107?color=ffc107&logoColor=white" />
   </a>
-   <a href="https://replicate.com/cjwbw/deepseek-math-7b-instruct" target="_parent"><img src="https://replicate.com/cjwbw/deepseek-math-7b-instruct/badge" alt="Replicate"/></a> 
+   <a href="https://replicate.com/cjwbw/deepseek-math-7b-base" target="_parent"><img src="https://replicate.com/cjwbw/deepseek-math-7b-base/badge" alt="Replicate"/></a> 
 </div>
 
 <div align="center">
diff --git a/cog.yaml b/cog.yaml
index d93c702..75bddf9 100644
--- a/cog.yaml
+++ b/cog.yaml
@@ -12,4 +12,4 @@ build:
     - hf_transfer
 
 # predict.py defines how predictions are run on your model
-predict: "predict.py:Predictor"
+predict: "replicate/predict.py:Predictor"
diff --git a/predict.py b/replicate/predict.py
similarity index 100%
rename from predict.py
rename to replicate/predict.py
diff --git a/predict_instruct.py b/replicate/predict_instruct.py
similarity index 100%
rename from predict_instruct.py
rename to replicate/predict_instruct.py