bwconrad commited on
Commit
74d38e8
1 Parent(s): 11ef2f5

Fix wrong path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ for k, v in ckpt.items():
59
  model.load_state_dict(new_state_dict, strict=True)
60
 
61
  # Load label names
62
- labels = pd.read_csv("app/classid_classname.csv", names=["id", "name"])["name"].tolist()
63
  labels = [l.replace("_", " ").title() for l in labels] # Remove _ and capitalize
64
 
65
  # Run app
 
59
  model.load_state_dict(new_state_dict, strict=True)
60
 
61
  # Load label names
62
+ labels = pd.read_csv("classid_classname.csv", names=["id", "name"])["name"].tolist()
63
  labels = [l.replace("_", " ").title() for l in labels] # Remove _ and capitalize
64
 
65
  # Run app