PriyePrabhakar commited on
Commit
fd1d2e7
·
1 Parent(s): 0c717d3

Add application file with new chkpoint

Browse files
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
app.py CHANGED
@@ -186,7 +186,7 @@ class_names_2 = list(class_names_dict.keys())
186
  def load_model():
187
  # Load the trained model
188
  model = ImageNetClassifier.load_from_checkpoint(
189
- "logs/checkpoints/epoch=58-val_loss=1.46.ckpt",
190
  map_location="cuda" if torch.cuda.is_available() else "cpu",
191
  #lr=6.28E-02, # This parameter is required but won't be used for inference
192
  num_classes=1000 # Make sure this matches your trained model
 
186
  def load_model():
187
  # Load the trained model
188
  model = ImageNetClassifier.load_from_checkpoint(
189
+ "logs/checkpoints/epoch=59-val_loss=1.46.ckpt",
190
  map_location="cuda" if torch.cuda.is_available() else "cpu",
191
  #lr=6.28E-02, # This parameter is required but won't be used for inference
192
  num_classes=1000 # Make sure this matches your trained model
logs/.DS_Store CHANGED
Binary files a/logs/.DS_Store and b/logs/.DS_Store differ
 
logs/checkpoints/{epoch=58-val_loss=1.46.ckpt → epoch=59-val_loss=1.46.ckpt} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6759f10d8a0d9f99681446be1794a905b29d3d2f74a1f20a5ff1efb7efb80d7c
3
  size 307147814
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4533925de5f0c0d6ad5a5c116d3a6d7b62d8b03eb6fce450850018eea1732c6
3
  size 307147814