matrix-spam-ml

commit 2ab9e33188ee3291ffeb7b5bd1b2607cb4606599
parent 070af63942f7cf1c875befc5d5cf1e1fc466919b
Author: MTRNord <mtrnord1@gmail.com>
Date:   Tue, 11 Oct 2022 15:21:20 +0200

Fix listener?

Diffstat:
M
crates/model_server/src/main.rs
 | 
11
++++++-----

1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/crates/model_server/src/main.rs b/crates/model_server/src/main.rs
@@ -7,6 +7,7 @@ use linkify::LinkFinder;
 use once_cell::sync::OnceCell;
 use serde::{Deserialize, Serialize};
 use std::io::Write;
+use std::net::{Ipv4Addr, Ipv6Addr};
 use std::{fs::OpenOptions, net::SocketAddr};
 use tensorflow::{Graph, SavedModelBundle, SessionOptions, SessionRunArgs, Tensor};
 use tracing::{error, info};
@@ -43,12 +44,12 @@ async fn main() -> Result<()> {
         .route("/submit", post(submit))
         .route("/submit_review", post(submit_for_review));
 
-    let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
-    info!("listening on {}", addr);
-    axum::Server::bind(&addr)
+    let all_v6 = SocketAddr::new(Ipv6Addr::UNSPECIFIED.into(), 3000);
+
+    info!("listening on port 3000");
+    axum::Server::bind(&all_v6)
         .serve(app.into_make_service())
-        .await
-        .unwrap();
+        .await?;
     Ok(())
 }