diff --git a/fly.toml b/fly.toml index 4e6e097..8aeebaf 100644 --- a/fly.toml +++ b/fly.toml @@ -4,7 +4,7 @@ image = "ghcr.io/fly-apps/cog-whisper:latest" [experimental] - cmd = ["./start.sh"] + cmd = ["python -m cog.server.http"] [env] COG_LOG_LEVEL = "debug" @@ -33,7 +33,3 @@ source = 'data' destination = '/src/weights' initial_size = '10G' - -[[files]] - guest_path = "/src/start.sh" - local_path = "./start.sh" diff --git a/predict.py b/predict.py index 2ebec5f..6d9ea1a 100644 --- a/predict.py +++ b/predict.py @@ -11,7 +11,7 @@ from whisper.utils import format_timestamp MODEL_CACHE = "weights" -BASE_URL = f"https://weights.replicate.delivery/default/whisper-v3/{MODEL_CACHE}/" +BASE_URL = f"https://fly.storage.tigris.dev/weights/models/whisper/" class ModelOutput(BaseModel): diff --git a/start.sh b/start.sh deleted file mode 100755 index a1b5523..0000000 --- a/start.sh +++ /dev/null @@ -1,10 +0,0 @@ -#!/bin/bash - -if [ -f /src/weights/large-v3.pt ] -then - echo "Weights already exist" -else - /src/scripts/get_weights.sh -fi - -python -m cog.server.http