Create main.rs

main.rs

@@ -0,0 +1,22 @@
+use axum::{routing::get, Router};
+use std::net::SocketAddr;
+
+#[tokio::main]
+async fn main() {
+    // Build our application with a single route
+    let app = Router::new()
+        .route("/", get(|| async { "Rust is running at Warp Speed on Hugging Face!" }))
+        .route("/health", get(|| async { "OK" }));
+
+    // HF Spaces default port is 7860
+    let port = std::env::var("PORT")
+        .unwrap_or_else(|_| "7860".to_string())
+        .parse::<u16>()
+        .unwrap();
+
+    let addr = SocketAddr::from(([0, 0, 0, 0], port));
+    println!("🚀 Server started on {}", addr);
+
+    let listener = tokio::net::TcpListener::bind(addr).await.unwrap();
+    axum::serve(listener, app).await.unwrap();
+}