From dc72f37e5fdc58f67d06911f9fc914a0ed44c883 Mon Sep 17 00:00:00 2001
From: Daniel Gatis <danielgatis@gmail.com>
Date: Sat, 12 Sep 2020 09:29:22 -0300
Subject: [PATCH] refactoring

---
 setup.py                | 2 +-
 src/rembg/cmd/server.py | 9 +++++----
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/setup.py b/setup.py
index d63ab22..76a63bc 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@ with open("requirements.txt") as f:
 
 setup(
     name="rembg",
-    version="1.0.9",
+    version="1.0.10",
     description="Remove image background",
     long_description=long_description,
     long_description_content_type="text/markdown",
diff --git a/src/rembg/cmd/server.py b/src/rembg/cmd/server.py
index da113b8..fa1bdfc 100644
--- a/src/rembg/cmd/server.py
+++ b/src/rembg/cmd/server.py
@@ -13,11 +13,8 @@ app = Flask(__name__)
 
 @app.route('/', methods=['GET', 'POST'])
 def index():
-    model = request.args.get("model", type=str, default="u2net")
-    if model not in ("u2net", "u2netp"):
-        return {"error": "invalid query param 'model'"}, 400
-
     file_content = ''
+
     if request.method == 'POST':
         if 'file' not in request.files:
             return {"error": "missing post form param 'file'"}, 400
@@ -34,6 +31,10 @@ def index():
     if file_content == '':
         return {"error": "File content is empty"}, 400
 
+    model = request.args.get("model", type=str, default="u2net")
+    if model not in ("u2net", "u2netp"):
+        return {"error": "invalid query param 'model'"}, 400
+
     try:
         return send_file(
             BytesIO(remove(file_content, model)),