From 60f4f059accf7f986f24dc1c6b06941c0232db2a Mon Sep 17 00:00:00 2001 From: Abhishek Thakur Date: Thu, 25 Apr 2024 15:15:41 +0200 Subject: [PATCH] redirect fi --- src/autotrain/oauth.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/autotrain/oauth.py b/src/autotrain/oauth.py index f24407a8c1..cc05b92ba5 100644 --- a/src/autotrain/oauth.py +++ b/src/autotrain/oauth.py @@ -97,8 +97,6 @@ async def auth(request: fastapi.Request) -> RedirectResponse: request.session["oauth_info"] = oauth_info return _redirect_to_target(request) - request.session["oauth_info"] = oauth_info - return _redirect_to_target(request) def _generate_redirect_uri(request: fastapi.Request) -> str: @@ -118,6 +116,6 @@ def _generate_redirect_uri(request: fastapi.Request) -> str: def _redirect_to_target(request: fastapi.Request, default_target: str = "/") -> RedirectResponse: - # target = request.query_params.get("_target_url", default_target) - target = "https://huggingface.co/spaces/" + os.environ.get("SPACE_ID") + target = request.query_params.get("_target_url", default_target) + # target = "https://huggingface.co/spaces/" + os.environ.get("SPACE_ID") return RedirectResponse(target)