Update app.py

app.py

@@ -72,10 +72,10 @@ def gen_sources(deepfake_img):
     ##----------------------Operation-------------------------------------------------
     with torch.no_grad():
         # Load and preprocess input image
-        img = Image.open(deepfake_img).convert('RGB')
-        segimg = Image.open(deepfake_seg).convert('RGB')
-        df_img = transform(img).unsqueeze(0).to(device)  # Shape: (1, 3, 256, 256)
-        seg_img = transform(segimg).unsqueeze(0).to(device)
+        #img = Image.open(deepfake_img).convert('RGB')
+        #segimg = Image.open(deepfake_seg).convert('RGB')
+        df_img = transform(deepfake_img.convert('RGB')).unsqueeze(0).to(device)  # Shape: (1, 3, 256, 256)
+        seg_img = transform(deepfake_seg).unsqueeze(0).to(device)
         
         # Calculate quantized_block for all images
         z_df, _, _ = model_vaq_f.encode(df_img)