From ec53e9ec8da19d4d8cbe2d9b7c54e631ef1b0252 Mon Sep 17 00:00:00 2001
From: russian_proger <sagitov.2806@mail.ru>
Date: Sun, 31 May 2026 13:07:08 +0300
Subject: [PATCH] ref: use uploaded weights

---
 gan.py             | 2 +-
 sian_similarity.py | 7 ++-----
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/gan.py b/gan.py
index 201051b..1866468 100644
--- a/gan.py
+++ b/gan.py
@@ -16,7 +16,7 @@ from vision_chunk import VisionChunk
 
 ROOT_DIR = Path(__file__).resolve().parent
 MODEL_FILE = ROOT_DIR / "models" / "GAN" / "src" / "model.py"
-DEFAULT_CHECKPOINT_PATH = ROOT_DIR / "models" / "GAN" / "runs" / "checkpoints" / "best.pth"
+DEFAULT_CHECKPOINT_PATH = ROOT_DIR / "models" / "TrainedWeights" / "GAN.pth"
 
 IMAGE_SIZE = (256, 256)
 CHECKPOINT_ENV = "GAN_CHECKPOINT"
diff --git a/sian_similarity.py b/sian_similarity.py
index 2ca2622..d282bbe 100644
--- a/sian_similarity.py
+++ b/sian_similarity.py
@@ -17,11 +17,8 @@ MODEL_FILE = ROOT_DIR / "models" / "SiaN-similarity" / "model.py"
 DEFAULT_CHECKPOINT_PATH = (
     ROOT_DIR
     / "models"
-    / "SiaN-similarity"
-    / "runs"
-    / "gan_training"
-    / "checkpoints"
-    / "best_model.pt"
+    / "TrainedWeights"
+    / "SiaN-Similarity.pt"
 )
 
 IMAGE_SIZE = (256, 256)