Update app.py

app.py

@@ -10,6 +10,7 @@ app.secret_key = os.urandom(24)
 ee_model = None
 ee_tokenizer = None
 ee_config = None
+ee_model_name = None
 
 SPACE_HOST = os.environ.get("SPACE_HOST", "")
 SPACE_URL = f"https://{SPACE_HOST}" if SPACE_HOST else "http://localhost:7860"
@@ -17,7 +18,7 @@ SPACE_URL = f"https://{SPACE_HOST}" if SPACE_HOST else "http://localhost:7860"
 
 @app.route("/", methods=["GET", "POST"])
 def index():
-    global ee_model, ee_tokenizer, ee_config
+    global ee_model, ee_tokenizer, ee_config, ee_model_name
 
     if request.method == "POST":
         ee_model_name = request.form["ee_model_name"].strip()