Spaces:

AravindKumarRajendran
/

ResNetonImageNet

Sleeping

AravindKumarRajendran commited on Dec 31, 2024

Commit

78d57af

1 Parent(s): da4ecbf

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -67,7 +67,7 @@ def load_model(model_path):
     model.eval()  # Set the model to evaluation mode
     return model
-model_path = "models/resnet50.pth"
 label_path = 'classes.json'  # Path to the class index mapping
 model = load_model(model_path)
 class_labels = load_class_labels(label_path)

     model.eval()  # Set the model to evaluation mode
     return model
+model_path = "models/resnet_50.pth"
 label_path = 'classes.json'  # Path to the class index mapping
 model = load_model(model_path)
 class_labels = load_class_labels(label_path)