Update app.py

app.py

@@ -38,6 +38,7 @@ classifier = PyTorchClassifier(
 # 이미지 전처리 함수
 def preprocess_image(image):
     transform = transforms.Compose([
+        transforms.Resize((224, 224)),
         transforms.ToTensor(),
         transforms.Normalize(mean=[0.485, 0.456, 0.406],
                              std=[0.229, 0.224, 0.225])
@@ -69,7 +70,7 @@ def generate_adversarial_image(image, eps_value):
     return adv_image_pil
 
 # 워터마크 삽입 함수
-def apply_watermark(image_pil, wm_text="123", password_img=123, password_wm=456):
+def apply_watermark(image_pil, wm_text="텍스트 삽입", password_img=000, password_wm=000):
     bwm = WaterMark(password_img=password_img, password_wm=password_wm)
 
     # 이미지 바이트 데이터를 임시 파일로 저장
@@ -97,7 +98,10 @@ def apply_watermark(image_pil, wm_text="123", password_img=123, password_wm=456)
 def process_image(image, eps_value, wm_text, password_img, password_wm):
     # 적대적 이미지 생성
     adv_image = generate_adversarial_image(image, eps_value)
-    watermarked_image = apply_watermark(image, wm_text, int(password_img), int(password_wm))
+    
+    # 적대적 이미지에 워터마크 삽입
+    watermarked_image = apply_watermark(adv_image, wm_text, int(password_img), int(password_wm))
+
     return watermarked_image
 
 # Gradio 인터페이스 정의