Updated model loading to use new checkpoint file name for CycleGAN model
Browse files- src/app.py +1 -1
- src/cyclegan_model_v3.pt +3 -0
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_v3.pt")
|
| 116 |
|
| 117 |
if not os.path.exists(model_path):
|
| 118 |
raise FileNotFoundError(f"Model file not found: {model_path}")
|
src/cyclegan_model_v3.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f24c136b4cf79c30cc2da4df7431d6ab1b4855644602073e3f74e9e7deea08e
|
| 3 |
+
size 113196781
|