@@ -27,6 +27,8 @@ class Prunaai(SimpleService):
     @classmethod
     def predict_price(cls, content: str, file_exists: bool, info: dict[str, Any]) -> Decimal | None:
         version = info['version']
+        if file_exists:
+            version = 'p-image-edit'
         return cls.TOKENS_COST[version]
 
     def save_results(self, content: str, t: timedelta, image_url: str, save: bool = True) -> list[Message]: