Update app.py

app.py

@@ -581,7 +581,7 @@ def test_single_image(generator_path, input_image_path):
     generator.eval()
 
     # Preprocess the input image
-    input_image = preprocess_input(input_image_path).to(device)
+input_image = preprocess_input(input_image_path).to(device)
 
     # Perform inference
     with torch.no_grad():
@@ -593,4 +593,4 @@ def test_single_image(generator_path, input_image_path):
         plt.show()
 
 
-    test_single_image('pretrained_model_064.pt','check.jpeg')
+test_single_image('pre_trained_model_064.pt',uploaded_image_data)