modify type answer

main.py

@@ -39,7 +39,7 @@ def safe_transform(encoder, value):
     if value in encoder.classes_:
         return encoder.transform([value])[0]
     else:
-        return 0  # Special code for unknown values
+        return -1  # Special code for unknown values
 
 class Msg(BaseModel):
     msg: str
@@ -129,7 +129,7 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
     prediction = model.predict(data)
     output = prediction[0]
 
-    output_text = "Legitimate" if output == 1 else "Phishing"
+    output_text = 1 if output == 1 else 0
 
     # Render index.html with prediction results
     json_compatible_resp_data = jsonable_encoder(Resp(url=requess.url, label=output_text))