Spaces:

Aravindan
/

butterfly_classification

Runtime error

Aravindan commited on Aug 6, 2022

Commit

981c905

1 Parent(s): 1d1e250

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ we_are = ['INDRA SWALLOW',
  'PAINTED LADY']
 def classify(image_):
-  model = model.eval()
   image = Image.open(image_)
   image = image_transforms(image).float().to(device)
   image = image.unsqueeze(0)

  'PAINTED LADY']
 def classify(image_):
+  model = new_model.eval()
   image = Image.open(image_)
   image = image_transforms(image).float().to(device)
   image = image.unsqueeze(0)