update server.py

server.py

@@ -172,10 +172,17 @@ def create_interface():
             allowed_paths=["cache", "css", "extensions", "js"]
         )
 
-if __name__ == "__main__":
+def download_model(model_name):
+    logger.info("Downloading " + model_name)
+    os.system(r"python3 download-model.py " + model_name)
+    logger.info("Download completed " + model_name)
+
 
+if __name__ == "__main__":
+    download_model("google/flan-t5-xl")
+    download_model("meta-llama/Meta-Llama-3-8B-Instruct")
     logger.info("Starting Text generation web UI")
-    logger.info("Need to download models...")
+    
     do_cmd_flags_warnings()
 
     # Load custom settings