Update app.py

app.py

@@ -24,7 +24,7 @@ def inference(img):
             print("No image received!")
             return None
         import sys
-        sys.argv = ['--model',  '--dataroot', '/home/data/luhaoxiang/wby/cyclegan/img/', '--num_test', '1',  '--no_dropout']
+        sys.argv = ['--model',  '--dataroot', './data/', '--num_test', '1',  '--no_dropout']
 
         # Load options and set them up
         opt = TestOptions().parse() 
@@ -49,7 +49,7 @@ def inference(img):
         img_tensor = img_tensor.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))  # Move to GPU if available
         
         # Prepare data for the model
-        data = {'A':img_tensor,'A_paths':'/home/data/luhaoxiang/wby/cyclegan/img/'}
+        data = {'A':img_tensor,'A_paths':'./data/'}
         model.set_input(data)
         model.test()