Update dockerfile

Dockerfile

@@ -9,6 +9,8 @@ RUN apt-get update && \
     wget \
     curl \
     # python build dependencies \
+    python3 \
+    pip \
     build-essential \
     libssl-dev \
     zlib1g-dev \
@@ -29,18 +31,14 @@ ENV HOME=/home/user \
     PATH=/home/user/.local/bin:${PATH}
 WORKDIR ${HOME}/app
 
-RUN curl https://pyenv.run | bash
-ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
-ARG PYTHON_VERSION=3.10.13
-RUN pyenv install ${PYTHON_VERSION} && \
-    pyenv global ${PYTHON_VERSION} && \
-    pyenv rehash && \
-    pip install --no-cache-dir -U pip setuptools wheel && \
-    pip install "huggingface-hub" "hf-transfer" "gradio[oauth]>=4.28.0" "gradio_huggingfacehub_search==0.0.7" "APScheduler"
+RUN pip3 install huggingface_hub transformers sentencepiece hf-transfer gradio[oauth]>=4.28.0 gradio_huggingfacehub_search==0.0.7 APScheduler
 
 COPY --chown=1000 . ${HOME}/app
 RUN git clone https://github.com/ggerganov/llama.cpp
-RUN pip install -r llama.cpp/requirements.txt
+
+RUN cd llama.cpp && \
+    pip3 install -r requirements.txt && \
+    make -j
 
 ENV PYTHONPATH=${HOME}/app \
     PYTHONUNBUFFERED=1 \
@@ -54,4 +52,4 @@ ENV PYTHONPATH=${HOME}/app \
     SYSTEM=spaces
 
 
-ENTRYPOINT /bin/sh start.sh
+CMD ["python", "app.py"]

app.py

@@ -1,7 +1,13 @@
 import gradio as gr
 import subprocess
+import os
 
 from huggingface_hub import HfApi, snapshot_download
+from gradio_huggingfacehub_search import HuggingfaceHubSearch
+
+from apscheduler.schedulers.background import BackgroundScheduler
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
 api = HfApi()
 
@@ -12,7 +18,7 @@ def process_model(model_id: str, file_path: str, key: str, value: str, hf_token)
 
     FILE_NAME = file_path.split("/")[-1]
 
-    snapshot_download(
+    api.snapshot_download(
         repo_id=model_id,
         allow_patterns=file_path,
         local_dir=f"{MODEL_NAME}",
@@ -50,3 +56,17 @@ iface = gr.Interface(
 
 # Launch the interface
 iface.launch(debug=True)
+
+
+def restart_space():
+    HfApi().restart_space(
+        repo_id="bartowski/gguf-metadata-updated", token=HF_TOKEN, factory_reboot=True
+    )
+
+
+scheduler = BackgroundScheduler()
+scheduler.add_job(restart_space, "interval", seconds=21600)
+scheduler.start()
+
+# Launch the interface
+demo.queue(default_concurrency_limit=1, max_size=5).launch(debug=True, show_api=False)