Update app.py

app.py

@@ -20,6 +20,7 @@ import mlx_lm
 from mlx_lm import convert
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
+SPACE_HOST = os.environ.get("SPACE_HOST")
 
 # I'm not sure if we need to add more stuff here
 QUANT_PARAMS = {
@@ -187,7 +188,7 @@ iface = gr.Interface(
 # Create Gradio interface
 with gr.Blocks(css=css) as demo:
     gr.Markdown("You must be logged in to use MLX-my-repo.")
-    gr.LoginButton()
+    gr.LoginButton(redirect=f"https://{SPACE_HOST}/login/callback")
 
     iface.render()