Spaces:
Runtime error
Runtime error
wayandadang
commited on
Commit
•
0d173bb
1
Parent(s):
d5a4cb8
Update app.py
Browse files
app.py
CHANGED
@@ -147,7 +147,7 @@ uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["jpg", "jpe
|
|
147 |
image_url = st.sidebar.text_input("Or enter image URL...")
|
148 |
|
149 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
150 |
-
model = load_model('weights/
|
151 |
|
152 |
img = None
|
153 |
|
|
|
147 |
image_url = st.sidebar.text_input("Or enter image URL...")
|
148 |
|
149 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
150 |
+
model = load_model('weights/best_model_VGG16_KAN_97.pth', device)
|
151 |
|
152 |
img = None
|
153 |
|