Update app.py

app.py

@@ -8,6 +8,11 @@ model_id = "sshleifer/tiny-gpt2"
 #deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B
  
 #model_id = "remi/bertabs-finetuned-extractive-abstractive-summarization"
+import shutil
+total, used, free = shutil.disk_usage("/")
+print("Total: %.2f GB" % (total / (2**30)))
+print("Used: %.2f GB" % (used / (2**30)))
+print("Free: %.2f GB" % (free / (2**30)))
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)