Spaces:

wayandadang
/

VGG16-KAN-for-Cat-and-Dog-Classification

Runtime error

wayandadang commited on Jun 18, 2024

Commit

0d173bb

verified ·

1 Parent(s): d5a4cb8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -147,7 +147,7 @@ uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["jpg", "jpe
 image_url = st.sidebar.text_input("Or enter image URL...")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = load_model('weights/best_model_vgg16_KAN.pth', device)
 img = None

 image_url = st.sidebar.text_input("Or enter image URL...")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = load_model('weights/best_model_VGG16_KAN_97.pth', device)
 img = None