soiz1 commited on
Commit
ea45c72
·
verified ·
1 Parent(s): 3b5feb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -204,6 +204,10 @@ def load_model():
204
  authors.append(model_author)
205
  model_path = f"/weights/{category_folder}/{character_name}/{model_name}"
206
  import pickle
 
 
 
 
207
  cpt = torch.load(f"/weights/{category_folder}/{character_name}/{model_name}", map_location="cpu", pickle_module=pickle)
208
  model_version = cpt.get("version", "v1")
209
  print(f"Indexed model {model_title} by {model_author} ({model_version})")
 
204
  authors.append(model_author)
205
  model_path = f"/weights/{category_folder}/{character_name}/{model_name}"
206
  import pickle
207
+ path = f"/weights/{category_folder}/{character_name}/{model_name}"
208
+
209
+ if not os.path.exists(path):
210
+ print(f"Model file not found: {path}")
211
  cpt = torch.load(f"/weights/{category_folder}/{character_name}/{model_name}", map_location="cpu", pickle_module=pickle)
212
  model_version = cpt.get("version", "v1")
213
  print(f"Indexed model {model_title} by {model_author} ({model_version})")