Update main.py

main.py

@@ -1,8 +1,11 @@
+import os
 import gradio as gr
 import helper
 
-# Create the Gradio interface using helper.registry
+USERS = os.environ.get("USERS")
+PWD = os.environ.get("PWD")
+
 app = helper.registry(name="marco-o1")
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch(auth=[USERS, PWD])