use_hf_storage

Dockerfile

@@ -5,6 +5,9 @@ WORKDIR /app
 
 COPY . .
 
+# This is only for building docker in huggingface spaces
+ENV HF_HOME=/data/.huggingface
+
 RUN conda env create -f environment_sate_0.11.yml
 
 RUN mkdir -p /app/session_data && chown -R user:user /app/session_data

modal_deploy.py

@@ -0,0 +1,20 @@
+import modal, subprocess, signal
+
+app = modal.App("sate-api")
+
+image = modal.Image.from_dockerfile("Dockerfile")
+
+@app.function(
+    image=image,
+    min_containers=1,
+    timeout=1000,
+    gpu="A10G",
+)
+
+@modal.web_server(7860)
+def run_server():
+    subprocess.Popen([
+        "conda", "run", "--no-capture-output", "-n", "SATE",
+        "python", "main_socket.py", "--host", "0.0.0.0", "--port", "7860"
+    ])
+    signal.pause()