Update app.py

app.py

@@ -44,9 +44,10 @@ logger.debug("done load")
 
 model_path = model.config._dict['model_name_or_path']
 logger.debug(f"{model_path=}")
-model_size_gb = os.path.getsize(model_path) / (1024 * 1024 * 1024)
 
-print(f"The model '{model_name}' is taking approximately {model_size_gb:.2f} GB of disk space.")
+model_size_gb = Path(model_path).stat().st_size / 2**30
+
+logger.info(f"{model_name=} {model_size_gb=:.2f} GB")
 
 # with gr.Blocks() as demo:
 #     chatbot = gr.Chatbot()