fix: python download script

Dockerfile

@@ -17,35 +17,36 @@ RUN pip install huggingface_hub
 # Create models directory
 RUN mkdir -p /models
 
-# Download models and sample images from seamo-ai/marina-species-v1
-RUN python3 << 'EOF'
-from huggingface_hub import hf_hub_download
-import os
-
-# Download model files
-print("Downloading model files...")
-hf_hub_download('seamo-ai/marina-species-v1', 'marina-benthic-33k.pt', local_dir='/models', local_dir_use_symlinks=False)
-hf_hub_download('seamo-ai/marina-species-v1', 'marina-benthic-33k.names', local_dir='/models', local_dir_use_symlinks=False)
-
-# Download all sample images
-print("Downloading sample images...")
-sample_images = [
-    'crab.png', 'fish.png', 'fish_2.png', 'fish_3.png', 'fish_4.png', 'fish_5.png',
-    'flat_fish.png', 'flat_red_fish.png', 'jelly.png', 'jelly_2.png', 'jelly_3.png',
-    'puff.png', 'red_fish.png', 'red_fish_2.png', 'scene.png', 'scene_2.png',
-    'scene_3.png', 'scene_4.png', 'scene_5.png', 'scene_6.png', 'soft_coral.png',
-    'starfish.png', 'starfish_2.png'
-]
-
-for img in sample_images:
-    try:
-        hf_hub_download('seamo-ai/marina-species-v1', f'images/{img}', local_dir='/models', local_dir_use_symlinks=False)
-        print(f'Downloaded {img}')
-    except Exception as e:
-        print(f'Failed to download {img}: {e}')
-
-print("Download complete!")
-EOF
+# Create download script
+RUN echo 'from huggingface_hub import hf_hub_download\n\
+import os\n\
+\n\
+# Download model files\n\
+print("Downloading model files...")\n\
+hf_hub_download("seamo-ai/marina-species-v1", "marina-benthic-33k.pt", local_dir="/models", local_dir_use_symlinks=False)\n\
+hf_hub_download("seamo-ai/marina-species-v1", "marina-benthic-33k.names", local_dir="/models", local_dir_use_symlinks=False)\n\
+\n\
+# Download sample images\n\
+print("Downloading sample images...")\n\
+sample_images = [\n\
+    "crab.png", "fish.png", "fish_2.png", "fish_3.png", "fish_4.png", "fish_5.png",\n\
+    "flat_fish.png", "flat_red_fish.png", "jelly.png", "jelly_2.png", "jelly_3.png",\n\
+    "puff.png", "red_fish.png", "red_fish_2.png", "scene.png", "scene_2.png",\n\
+    "scene_3.png", "scene_4.png", "scene_5.png", "scene_6.png", "soft_coral.png",\n\
+    "starfish.png", "starfish_2.png"\n\
+]\n\
+\n\
+for img in sample_images:\n\
+    try:\n\
+        hf_hub_download("seamo-ai/marina-species-v1", f"images/{img}", local_dir="/models", local_dir_use_symlinks=False)\n\
+        print(f"Downloaded {img}")\n\
+    except Exception as e:\n\
+        print(f"Failed to download {img}: {e}")\n\
+\n\
+print("Download complete!")' > /tmp/download_models.py
+
+# Run the download script
+RUN python3 /tmp/download_models.py
 
 # Stage 2: Build the application
 FROM python:3.10-slim