init push

app.py

@@ -433,7 +433,6 @@ with gr.Blocks(
                 label="Preview",
                 columns=4,
                 height="300px",
-                show_download_button=True,
                 object_fit="contain",
                 preview=True,
             )

setup_envs.sh

@@ -0,0 +1,39 @@
+#!/bin/bash
+
+# Script to verify PyTorch dependencies in Docker container
+# This script checks existing PyTorch/numpy versions and modifies requirements.txt
+# to avoid reinstalling them, using the base image versions instead.
+
+set -e
+
+echo "=== Starting dependency verification script ==="
+
+# Check existing PyTorch and numpy
+echo "=== Checking existing PyTorch and numpy ==="
+python3 -c "import torch; import torchvision; import numpy; print(f'Existing torch: {torch.__version__}, torchvision: {torchvision.__version__}, numpy: {numpy.__version__}')" || echo "check failed"
+
+# Verify requirements.txt exists
+if [ ! -f "requirements.txt" ]; then
+    echo "ERROR: requirements.txt not found in current directory"
+    exit 1
+fi
+
+echo "=== Original requirements.txt ==="
+cat requirements.txt
+
+# Modify requirements to remove torch, torchvision, and numpy (to use system versions)
+echo "=== Modifying requirements.txt to remove torch, torchvision, and numpy ==="
+sed -i '/^torch==/d' requirements.txt
+sed -i '/^torch>=/d' requirements.txt
+sed -i '/^torch$/d' requirements.txt
+sed -i '/^torchvision/d' requirements.txt
+sed -i '/^numpy/d' requirements.txt
+
+echo "=== Modified requirements.txt ==="
+cat requirements.txt
+
+# Install modified requirements
+echo "=== Installing modified requirements ==="
+pip install --no-cache-dir --ignore-installed blinker -r requirements.txt
+
+echo "=== Dependency verification completed successfully ==="