Create startup.sh

startup.sh

@@ -0,0 +1,32 @@
+#!/bin/bash
+
+# Set the model name and URL
+MODEL_NAME="reswapper-429500.pth"  # Or the name of your model file
+MODEL_URL="https://huggingface.co/somanchiu/reswapper/resolve/main/reswapper-429500.pth" # Replace with your model's URL
+
+# Check if the model already exists. If so, skip the download
+if [ ! -f "$MODEL_NAME" ]; then
+  echo "Downloading model $MODEL_NAME from $MODEL_URL..."
+  wget "$MODEL_URL" -O "$MODEL_NAME"
+  if [ $? -eq 0 ]; then # Check wget exit code (0 means success)
+    echo "Model $MODEL_NAME downloaded successfully."
+  else
+    echo "Error downloading model $MODEL_NAME. Check the URL and network connection."
+    exit 1 # Exit the script with an error code
+  fi
+else
+  echo "Model $MODEL_NAME already exists. Skipping download."
+fi
+
+
+# (Optional) Download other files if needed - follow the same pattern as above:
+# OTHER_FILE="other_file.txt"
+# OTHER_URL="https://example.com/other_file.txt"
+# if [ ! -f "$OTHER_FILE" ]; then
+#   wget "$OTHER_URL" -O "$OTHER_FILE"
+# fi
+
+# Start your Gradio app
+echo "Starting Gradio app..."
+python app.py
+