blanar commited on
Commit
bdf731b
Β·
1 Parent(s): 7a4b189

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('../dog')
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('../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))
 
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