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

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

Browse files
Files changed (1) hide show
  1. 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, "cyclegan_model_v3.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_v2.pt")
116
 
117
  if not os.path.exists(model_path):
118
  raise FileNotFoundError(f"Model file not found: {model_path}")