soiz1 commited on
Commit
d5f75d1
·
verified ·
1 Parent(s): 6245a11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -201,7 +201,7 @@ def load_model():
201
 
202
  print(f"Loading model from: {path}")
203
  try:
204
- cpt = torch.load(path, map_location="cpu", pickle_module=pickle)
205
  model_version = cpt.get("version", "v1")
206
  print(f"Indexed model {model_title} by {info.get('author', 'Unknown')} ({model_version})")
207
  except Exception as e:
 
201
 
202
  print(f"Loading model from: {path}")
203
  try:
204
+ cpt = torch.load(path, map_location="cpu")
205
  model_version = cpt.get("version", "v1")
206
  print(f"Indexed model {model_title} by {info.get('author', 'Unknown')} ({model_version})")
207
  except Exception as e: