Update app.py

app.py

@@ -24,7 +24,7 @@ main_model.classifier[3] = nn.Linear(num_ftrs, 2)
 #     nn.Linear(num_ftrs, 2)  # 2 classes: AI-generated Image, Real Image
 # )
 
-main_model.load_state_dict(torch.load('best_model6for_RESNET18_After_First_Half_training_part.pth', map_location=device, weights_only=True))  # Updated: weights_only=True
+main_model.load_state_dict(torch.load('best_model3_mobilenetv3_large.pth', map_location=device, weights_only=True))  # Updated: weights_only=True
 main_model = main_model.to(device)
 main_model.eval()