Browse Source

Merge pull request #149 from vkosnar/patch-1

Hotfix spaces in URL in GET requests
Daniel Gatis 3 years ago
parent
commit
ba8c87ac2d
1 changed files with 5 additions and 1 deletions
  1. 5 1
      src/rembg/cmd/server.py

+ 5 - 1
src/rembg/cmd/server.py

@@ -27,8 +27,12 @@ def index():
         url = request.args.get("url", type=str)
         if url is None:
             return {"error": "missing query param 'url'"}, 400
+        
+        url = unquote_plus(url)
+        if " " in url:
+            url = quote(url, safe="/:")
 
-        file_content = urlopen(unquote_plus(url)).read()
+        file_content = urlopen(url).read()
 
     if file_content == "":
         return {"error": "File content is empty"}, 400