Create app.py

app.py

@@ -0,0 +1,36 @@
+from pathlib import Path
+import gradio as gr
+from huggingface_hub import snapshot_download
+
+snapshot_download("tiiuae/falcon-7b")
+
+DATA_PATH = Path("/data")
+
+def get_storage():
+    files = [
+        {
+            "orig_name": file.name,
+            "name": file.resolve(),
+            "size": file.stat().st_size,
+            "data": None,
+            "is_file": True,
+        }
+        for file in DATA_PATH.glob("**/*")
+        if file.is_file()
+    ]
+    usage = sum([f['size'] for f in files])
+
+    return files, f"{usage/(1024.0 ** 3):.3f}GB"
+
+
+with gr.Blocks() as app:
+    with gr.Row():
+        with gr.Column():
+            btn = gr.Button("Run")
+        with gr.Column():
+            files = gr.Files(label="Files")
+            storage = gr.Text(label="Total Usage")
+    btn.click(get_storage, inputs=None, outputs=[files, storage], postprocess=False)
+
+# Files that you explicitly allow on allowed_paths
+app.launch(allowed_paths=["/data"])