Update app.py

app.py

@@ -80,7 +80,7 @@ def check_plagiarism(text):
         token_text = tokenizer.decode(token_id.unsqueeze(0))
         
         # If the token is part of the watermark, use a mean or max threshold on g_val if it's multi-element
-        if mask.item() and g_val.mean().item() > 0.5:  # Use .mean() to get a scalar value
+        if mask.item() and g_val.float().mean().item() > 0.5:  # Use .mean() to get a scalar value
             highlighted_text += f"<mark>{token_text}</mark>"  # Highlight watermarked content
         else:
             highlighted_text += token_text