diff --git a/cog.yaml b/cog.yaml
index 63ad6a4..46558c1 100644
--- a/cog.yaml
+++ b/cog.yaml
@@ -10,4 +10,4 @@ build:
   run:
     - pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
 
-predict: "cogrun.py:Predictor"
+predict: "replicate_predictor.py:ReplicatePredictor"
diff --git a/cogrun.py b/replicate_predictor.py
similarity index 95%
rename from cogrun.py
rename to replicate_predictor.py
index 0697bf2..eccea66 100644
--- a/cogrun.py
+++ b/replicate_predictor.py
@@ -6,7 +6,7 @@ from math import log2
 from cog import BasePredictor, Path, Input
 
 
-class Predictor(BasePredictor):
+class ReplicatePredictor(BasePredictor):
     def setup(self):
         self.model = MinDalle(is_mega=True)