Update app.py

app.py

@@ -99,7 +99,7 @@ def ant_genus_classification(image):
 
     print(image.shape)
     with torch.no_grad():
-        y_hat = model(x)
+        y_hat = model(image)
         y_hat = torch.nn.functional.softmax(y_hat, dim=1)
         preds = torch.argmax(y_hat, dim=1)