Update app.py

app.py

@@ -31,13 +31,15 @@ modelInfo = [
 
 # Download model files if not exist
 for model in modelInfo:
-    directory = model[0]
+    directory = model[0].replace(" ", "_")
     if not os.path.exists(directory):
         os.makedirs(directory)
-    if not os.path.exists(os.path.join(directory, "best_model.pth")):
+    model_path = os.path.join(directory, "best_model.pth")
+    config_path = os.path.join(directory, "config.json")
+    if not os.path.exists(model_path):
         print(f"|> Downloading model: {directory}")
         download_url(model[3] + model[1], directory, "best_model.pth")
-    if not os.path.exists(os.path.join(directory, "config.json")):
+    if not os.path.exists(config_path):
         print(f"|> Downloading config: {directory}")
         download_url(model[3] + model[2], directory, "config.json")