Update app.py

app.py

@@ -7,12 +7,17 @@ WATERMARK_MESSAGE = 0b101100111110110010010000011110111011000110011110
 # bin(x)[2:] gives bits of x as str, use int to convert them to 0/1
 WATERMARK_BITS = [int(bit) for bit in bin(WATERMARK_MESSAGE)[2:]]
 
+SDXL_MESSAGE=[bool(bit) for bit in bin(WATERMARK_MESSAGE)[2:]]
+
 def detect_watermark(image):
     decoder = WatermarkDecoder('bits', 48)
     watermark = decoder.decode(image, 'dwtDct')
-    print(watermark)
-    return "何もわかりませんでした。"
-    #return "この画像はdiffusersで作られた可能性があります。"
+    
+    if(all(watermark==SDXL_MESSAGE)):
+        return "この画像はdiffusersのSDXLで作られた可能性があります。"
+    else:
+        return "何もわかりませんでした。"
+
 gr.Interface(fn=detect_watermark,
              title="AI画像簡易チェックツール",
              description="指定の画像が画像生成AIによって作られたものかを簡易的に判定します。このツールで判定したことを決して断定的に用いないでください。",