Update Dockerfile

Dockerfile

@@ -1,26 +1,20 @@
-FROM h2oairelease/h2oai-llmstudio-app:latest
+FROM nvidia/cuda:12.1.0-base-ubuntu22.04
 
 USER root
-
-RUN mkdir -p /workspace/mount && \
-    chmod -R 777 /workspace/mount && \
-    chmod -R 777 /workspace
+RUN apt-get update && apt-get install -y curl bash && \
+    mkdir -p /workspace/mount && \
+    chmod -R 777 /workspace && \
+    useradd -m -u 1000 user
 
 USER 1000
-
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    H2O_LLM_STUDIO_WORKDIR=/workspace/mount \
-    HF_HOME=/workspace/mount/huggingface \
-    TRITON_CACHE_DIR=/workspace/mount/.triton/cache \
-    H2O_WAVE_DATA_DIR=/workspace/mount/wave_data \
-    H2O_WAVE_MAX_REQUEST_SIZE=25MB \
-    H2O_WAVE_NO_LOG=True \
-    H2O_WAVE_PRIVATE_DIR="/download/@/workspace/mount/output/download" \
-    H2O_WAVE_ADDRESS=http://0.0.0.0:10101
+    PATH=/home/user/.local/bin:$PATH
 
-WORKDIR /workspace
+RUN curl -fsSL https://lmstudio.ai/install.sh | bash
 
-EXPOSE 10101
+ENV PATH="/home/user/.lmstudio/bin:$PATH"
+
+WORKDIR /workspace
+EXPOSE 7860
 
-CMD ["wave", "run", "llm_studio.app"]
+CMD ["sh", "-c", "lms daemon up && lms server start --port 7860 --host 0.0.0.0"]