avoid re-installing pytorch

app.py

@@ -4,6 +4,7 @@ Any new model should implement the following functions:
 - load_model(model_path, img_size=640)
 - inference(model, image)
 """
+
 import os
 import glob
 import spaces
@@ -16,8 +17,11 @@ from utils import (
     FlaggedCounter,
 )
 from flagging import HuggingFaceDatasetSaver
+import numpy as np
+
 import install_private_repos
 from seavision import load_model, AHOY
+from seavision.utils.general import LOGGER
 
 
 TITLE = """
@@ -44,7 +48,15 @@ h1 {
 }
 """
 
-model = load_model("ahoy-RGB-b2")
+rng = np.random.default_rng(0xDEADBEEF)
+
+
+@spaces.GPU
+def load_ahoy() -> AHOY:
+    """Load and warmup the model."""
+    model = load_model("ahoy-RGB-b2")
+    return model
+
 
 @spaces.GPU
 def inference(model: AHOY, image):
@@ -52,6 +64,9 @@ def inference(model: AHOY, image):
     results = model(image)
     return results.draw(image, diameter=4)
 
+
+model = load_ahoy()
+
 # Flagging
 dataset_name = "SEA-AI/crowdsourced-sea-images"
 hf_writer = HuggingFaceDatasetSaver(get_token(), dataset_name)
@@ -95,7 +110,7 @@ with gr.Blocks(theme=theme, css=css, title="SEA.AI Vision Demo") as demo:
 
     # event listeners
     img_url.change(load_image_from_url, [img_url], img_input)
-    submit.click(check_image, [img_input], show_api=False).success(
+    submit.click(check_image, [img_input], None, show_api=False).success(
         lambda image: inference(model, image),
         [img_input],
         img_output,

install_private_repos.py

@@ -3,9 +3,8 @@ import os
 GITHUB_TOKEN = os.getenv("GITHUB_TOKEN")
 if GITHUB_TOKEN is not None:
     try:
-        cmd = f'pip3 install "seavision[inference] @ git+https://{GITHUB_TOKEN}@github.com/SEA-AI/seavision"'
-        cmd += ' --upgrade-strategy only-if-needed'
+        cmd = f"pip3 install git+https://{GITHUB_TOKEN}@github.com/SEA-AI/seavision.git"
         os.system(cmd)
     except Exception as e:
         print(f"Error: {e}")
-        raise e
+        raise e

requirements.txt

@@ -1,2 +1,4 @@
 spaces
-pillow
+pillow
+onnx
+onnx2torch