Spaces:
Sleeping
Sleeping
Commit
·
067ff37
1
Parent(s):
de70513
update the model file
Browse files
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 = "
|
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 |
|