Updated model loading to use new checkpoint file name for CycleGAN model
Browse files- src/app.py +1 -1
src/app.py
CHANGED
|
@@ -112,7 +112,7 @@ class CycleGAN(nn.Module):
|
|
| 112 |
@st.cache_resource
|
| 113 |
def load_model():
|
| 114 |
model_dir = os.path.dirname(os.path.abspath(__file__))
|
| 115 |
-
model_path = os.path.join(model_dir, "
|
| 116 |
|
| 117 |
if not os.path.exists(model_path):
|
| 118 |
raise FileNotFoundError(f"Model file not found: {model_path}")
|
|
|
|
| 112 |
@st.cache_resource
|
| 113 |
def load_model():
|
| 114 |
model_dir = os.path.dirname(os.path.abspath(__file__))
|
| 115 |
+
model_path = os.path.join(model_dir, "cyclegan_model_v2.pt")
|
| 116 |
|
| 117 |
if not os.path.exists(model_path):
|
| 118 |
raise FileNotFoundError(f"Model file not found: {model_path}")
|