Spaces:
Build error
Build error
Fix wrong path
Browse files
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("
|
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
|