satyanayak commited on
Commit
067ff37
·
1 Parent(s): de70513

update the model file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def convert_state_dict(state_dict):
51
  def load_model():
52
  try:
53
  repo_id = "satyanayak/imagenet-resnet50-composer-model"
54
- filename = "pytorch_model_latest.bin"
55
 
56
  print(f"Attempting to load model from {repo_id}/{filename}")
57
 
 
51
  def load_model():
52
  try:
53
  repo_id = "satyanayak/imagenet-resnet50-composer-model"
54
+ filename = "model.pt"
55
 
56
  print(f"Attempting to load model from {repo_id}/{filename}")
57