Spaces:

victor
/

caca23232

Sleeping

App Files Files Community

victor HF Staff commited on about 1 month ago

Commit

9ff39d9

1 Parent(s): f087778

use both model and bucket volumes

Browse files

Files changed (1) hide show

app.py +26 -14

app.py CHANGED Viewed

@@ -1,4 +1,5 @@
 import os
 from flask import Flask
 app = Flask(__name__)
@@ -8,22 +9,33 @@ def index():
     model_dir = "/model"
     data_dir = "/data"
-    def list_dir(path, max_files=50):
-        if not os.path.exists(path):
-            return f"<p><b>{path}</b> does not exist</p>"
-        entries = sorted(os.listdir(path))[:max_files]
-        if not entries:
-            return f"<p><b>{path}</b> is empty</p>"
-        items = "".join(f"<li>{e} ({os.path.getsize(os.path.join(path, e)):,} bytes)</li>"
-                        if os.path.isfile(os.path.join(path, e))
-                        else f"<li>{e}/</li>"
-                        for e in entries)
-        return f"<h2>{path}</h2><ul>{items}</ul>"
     return f"""<html><body style="font-family: -apple-system, sans-serif; max-width: 600px; margin: 40px auto;">
-<h1>Volume Mount Test</h1>
-{list_dir(model_dir)}
-{list_dir(data_dir)}
 </body></html>"""
 if __name__ == "__main__":

 import os
+from datetime import datetime
 from flask import Flask
 app = Flask(__name__)
     model_dir = "/model"
     data_dir = "/data"
+    # Write visit to bucket
+    os.makedirs(data_dir, exist_ok=True)
+    visits_path = os.path.join(data_dir, "visits.txt")
+    with open(visits_path, "a") as f:
+        f.write(f"{datetime.utcnow().isoformat()}Z\n")
+    with open(visits_path) as f:
+        visits = f.read().strip().split("\n")
+    # List model files
+    model_files = []
+    if os.path.exists(model_dir):
+        for e in sorted(os.listdir(model_dir)):
+            p = os.path.join(model_dir, e)
+            if os.path.isfile(p):
+                model_files.append(f"<li>{e} ({os.path.getsize(p):,} bytes)</li>")
+            else:
+                model_files.append(f"<li>{e}/</li>")
     return f"""<html><body style="font-family: -apple-system, sans-serif; max-width: 600px; margin: 40px auto;">
+<h1>Two Volumes Test</h1>
+<h2>/model <span style="color:#86868b; font-weight:400; font-size:14px;">(read-only, openai-community/gpt2)</span></h2>
+<ul>{"".join(model_files) if model_files else "<li>empty</li>"}</ul>
+<h2>/data <span style="color:#86868b; font-weight:400; font-size:14px;">(read-write, victor/caca bucket)</span></h2>
+<p>{len(visits)} visit{"s" if len(visits) != 1 else ""} recorded. Data persists across restarts.</p>
+<pre style="background:#f5f5f7; padding:12px; border-radius:8px; font-size:13px; max-height:200px; overflow:auto;">{chr(10).join(visits[-20:])}</pre>
 </body></html>"""
 if __name__ == "__main__":