Update app.py

app.py

@@ -15,7 +15,7 @@ def load_model():
         torch.nn.ReLU(),
         torch.nn.Linear(256, num_classes)
     )  
-    model.load_state_dict(torch.load(r"/kaggle/input/rice_epoch8/pytorch/default/1/best_model_epoch_8.pth", map_location=device), strict=False)
+    model.load_state_dict(torch.load("best_model_epoch_43.pth", map_location=device), strict=False)
     model = model.to(device)
     model.eval()
     return model