Update app.py

app.py

@@ -19,7 +19,7 @@ main_model.fc = nn.Sequential(
     nn.Linear(num_ftrs, 2)  # 2 classes: AI-generated Image, Real Image
 )
 
-main_model.load_state_dict(torch.load('best_model9.pth', map_location=device, weights_only=True))  # Updated: weights_only=True
+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 = main_model.to(device)
 main_model.eval()