nragrawal commited on
Commit
d6c38cc
·
1 Parent(s): 33566fb

Update username in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import sys
10
  def load_model():
11
  try:
12
  model = AutoModelForImageClassification.from_pretrained(
13
- "nragrawal/resnet-imagenet",
14
  trust_remote_code=True
15
  )
16
  model.eval()
 
10
  def load_model():
11
  try:
12
  model = AutoModelForImageClassification.from_pretrained(
13
+ "nragrawal/resnet-imagenet1k",
14
  trust_remote_code=True
15
  )
16
  model.eval()