Sckwoky commited on
Commit
a881650
·
1 Parent(s): c4e718e

Updated model loading to use new checkpoint file name for CycleGAN model

Browse files
Files changed (2) hide show
  1. src/app.py +1 -1
  2. 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, "cyclegan_model_v2.pt")
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