Update app.py

app.py

@@ -4,22 +4,33 @@ from huggingface_hub import snapshot_download
 
 snapshot_download("tiiuae/falcon-7b")
 
-DATA_PATH = "/data"
+DATA_PATH = Path("/data")
 
 def get_storage():
-  dir = Path(DATA_PATH)
-  files = [str(f) for f in dir.glob("**/*") if f.is_file()]
-  usage = sum([f.stat().st_size for f in dir.glob("**/*") if f.is_file()])
-  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, None, [files, storage])
-app.launch()
+    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=[str(DATA_PATH)])