Merge branch 'main' of https://huggingface.co/spaces/yifan0sun/BERTGradGraph

server.py

@@ -61,8 +61,6 @@ class PredModelRequest(BaseModel):
 def ping():
     return {"message": "pong"}
 
-
-
     
 def zip_if_needed(src_dir, zip_path):
     if os.path.exists(zip_path):
@@ -152,6 +150,17 @@ def copy_and_extract():
 
 
 
+@app.get("/data_check")
+def data_check():
+    with open("/data/marker.txt", "w") as f:
+        f.write("hello from server.py\n")
+        
+    files = os.listdir("/data")
+    return {
+        "message": "done",
+        "contents": files
+    }
+