Update install.sh

install.sh

@@ -1,13 +1,29 @@
 #!/bin/bash
 set -e
 
-echo "📦 Installing dependencies from requirements.txt..."
-pip install --no-cache-dir -r requirements.txt
+echo "🚀 Setting up Multimodal AI Environment"
+echo "--------------------------------------"
 
-echo "🔄 Forcing correct huggingface-hub version..."
-pip install --force-reinstall huggingface-hub==0.19.0
+# Create virtual environment
+python -m venv .venv
+source .venv/bin/activate
 
-echo "🔍 Checking installed huggingface-hub version..."
-pip show huggingface-hub | grep Version
+# Install core requirements
+pip install --upgrade pip
+pip install -r requirements.txt
 
-echo "✅ Installation complete."
+# Verify critical installations
+echo "✅ Installation Complete"
+echo "Installed Packages:"
+pip list | grep -E "torch|transformers|diffusers|gradio"
+
+# Space-specific optimizations
+if [ "$SPACES" = "1" ]; then
+    echo "🔧 Applying Space Optimizations"
+    mkdir -p model_cache
+    mkdir -p tmp
+    chmod -R 777 model_cache
+    chmod -R 777 tmp
+fi
+
+echo "🌐 Ready to launch!"