Spaces:
Sleeping
Sleeping
weights to folder
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ paw_dls = dls.dataloaders(train_df, batch_size=8, seed=12, device='cpu')
|
|
| 37 |
test = paw_dls.test_dl(train_df)
|
| 38 |
learn = cnn_learner(paw_dls, models.resnet50, pretrained=False, metrics=error_rate)
|
| 39 |
learn.to('cpu')
|
| 40 |
-
catanddog = learn.load('
|
| 41 |
|
| 42 |
st.title("Hot Dog? Or Not?")
|
| 43 |
|
|
@@ -56,11 +56,11 @@ if file_name is not None:
|
|
| 56 |
learn = Learner(paw_dls, model, loss_func = BCEWithLogitsLossFlat(), metrics=metric_rmse)
|
| 57 |
# learner = learner.load(f'/kaggle/input/puredog/pure-dog-{fold}')
|
| 58 |
if pred == 0:
|
| 59 |
-
learn.load('
|
| 60 |
pred = learn.predict(image)
|
| 61 |
print("Score for this doggy:", int(pred[2] * 100))
|
| 62 |
if pred == 1:
|
| 63 |
-
learn.load(f'
|
| 64 |
pred = learn.predict(image)
|
| 65 |
print(pred)
|
| 66 |
print("Score for this catto:", min(int(pred[2] * 100 + 35), 100))
|
|
|
|
| 37 |
test = paw_dls.test_dl(train_df)
|
| 38 |
learn = cnn_learner(paw_dls, models.resnet50, pretrained=False, metrics=error_rate)
|
| 39 |
learn.to('cpu')
|
| 40 |
+
catanddog = learn.load('dog')
|
| 41 |
|
| 42 |
st.title("Hot Dog? Or Not?")
|
| 43 |
|
|
|
|
| 56 |
learn = Learner(paw_dls, model, loss_func = BCEWithLogitsLossFlat(), metrics=metric_rmse)
|
| 57 |
# learner = learner.load(f'/kaggle/input/puredog/pure-dog-{fold}')
|
| 58 |
if pred == 0:
|
| 59 |
+
learn.load('pure-dog-5')
|
| 60 |
pred = learn.predict(image)
|
| 61 |
print("Score for this doggy:", int(pred[2] * 100))
|
| 62 |
if pred == 1:
|
| 63 |
+
learn.load(f'pure-cat-0')
|
| 64 |
pred = learn.predict(image)
|
| 65 |
print(pred)
|
| 66 |
print("Score for this catto:", min(int(pred[2] * 100 + 35), 100))
|
dog.pth β models/dog.pth
RENAMED
|
File without changes
|
pure-cat-0.pth β models/pure-cat-0.pth
RENAMED
|
File without changes
|
pure-dog-5.pth β models/pure-dog-5.pth
RENAMED
|
File without changes
|