feat: atualização do detector com otimizações para GPU T4

Dockerfile

@@ -36,23 +36,22 @@ RUN pip install --no-cache-dir -U pip setuptools wheel && \
     pip install --no-cache-dir --upgrade "tokenizers>=0.15.0" && \
     pip install --no-cache-dir --upgrade "git+https://github.com/huggingface/transformers.git"
 
-# Criar diretório de vídeos e cache
-RUN mkdir -p /code/videos /code/.cache/huggingface /code/.cache/torch && \
-    chmod -R 777 /code/.cache /code/videos
+# Criar diretório de vídeos
+RUN mkdir -p /code/videos && chmod -R 777 /code/videos
 
 # Configurar variáveis de ambiente
 ENV HOST=0.0.0.0 \
     PORT=7860 \
     PYTHONUNBUFFERED=1 \
     PYTHONPATH=/code \
-    TRANSFORMERS_CACHE=/code/.cache/huggingface \
-    TORCH_HOME=/code/.cache/torch \
+    TRANSFORMERS_CACHE=/tmp/huggingface \
+    TORCH_HOME=/tmp/torch \
     GRADIO_SERVER_NAME=0.0.0.0 \
     GRADIO_SERVER_PORT=7860 \
     SYSTEM=spaces \
     CUDA_VISIBLE_DEVICES=0 \
-    HUGGINGFACE_HUB_CACHE=/code/.cache/huggingface \
-    HF_HOME=/code/.cache/huggingface \
+    HUGGINGFACE_HUB_CACHE=/tmp/huggingface \
+    HF_HOME=/tmp/huggingface \
     TORCH_CUDA_ARCH_LIST="7.5" \
     MAX_WORKERS=2 \
     TRANSFORMERS_OFFLINE=0 \

deploy.sh

@@ -18,25 +18,18 @@ echo "🚀 Preparando deploy para GitHub..."
 echo "📦 Adicionando arquivos para GitHub..."
 git add .
 
-# Verificar se há alterações para commitar
-if [[ -n $(git status -s) ]]; then
-    echo "📝 Existem alterações para commitar no GitHub"
-    
-    # Fazer commit
-    git commit -m "$commit_message"
-    
-    # Force push para GitHub
-    echo "🚀 Forçando push para GitHub..."
-    git push -f origin main
-    
-    if [ $? -eq 0 ]; then
-        echo "✅ Deploy para GitHub concluído com sucesso!"
-    else
-        echo "❌ Erro durante o deploy para GitHub"
-        exit 1
-    fi
+# Fazer commit
+git commit -m "$commit_message"
+
+# Force push para GitHub
+echo "🚀 Forçando push para GitHub..."
+git push -f origin main
+
+if [ $? -eq 0 ]; then
+    echo "✅ Deploy para GitHub concluído com sucesso!"
 else
-    echo "✨ Workspace limpo, nenhuma alteração para GitHub"
+    echo "❌ Erro durante o deploy para GitHub"
+    exit 1
 fi
 
 # Deploy para Hugging Face
@@ -54,32 +47,26 @@ fi
 echo "📦 Adicionando todos os arquivos..."
 git add --all
 
-# Verificar se há alterações para o Hugging Face
-if [[ -n $(git status -s) ]]; then
-    echo "📝 Existem alterações para commitar no Hugging Face"
-    
-    # Fazer commit
-    git commit -m "$commit_message"
-    
-    echo "🚀 Enviando para Hugging Face Space..."
-    git push -f space main
-    
-    if [ $? -eq 0 ]; then
-        echo "✅ Deploy para Hugging Face concluído com sucesso!"
-        echo "🌐 Seu app estará disponível em alguns minutos em:"
-        echo "   https://huggingface.co/spaces/marcuscanhaco/weapon-detection-app"
-        echo ""
-        echo "⚠️ Lembre-se de verificar no Hugging Face Space se:"
-        echo "  1. O Space está configurado para usar GPU T4"
-        echo "  2. As variáveis de ambiente estão configuradas corretamente:"
-        echo "     - HUGGING_FACE_TOKEN"
-        echo "     - NOTIFICATION_EMAIL"
-        echo "     - SENDGRID_API_KEY"
-        echo "  3. Os requisitos de memória estão adequados"
-    else
-        echo "❌ Erro durante o deploy para Hugging Face"
-        exit 1
-    fi
+# Fazer commit
+git commit -m "$commit_message"
+
+# Force push para Hugging Face
+echo "🚀 Enviando para Hugging Face Space..."
+git push -f space main
+
+if [ $? -eq 0 ]; then
+    echo "✅ Deploy para Hugging Face concluído com sucesso!"
+    echo "🌐 Seu app estará disponível em alguns minutos em:"
+    echo "   https://huggingface.co/spaces/marcuscanhaco/weapon-detection-app"
+    echo ""
+    echo "⚠️ Lembre-se de verificar no Hugging Face Space se:"
+    echo "  1. O Space está configurado para usar GPU T4"
+    echo "  2. As variáveis de ambiente estão configuradas corretamente:"
+    echo "     - HUGGING_FACE_TOKEN"
+    echo "     - NOTIFICATION_EMAIL"
+    echo "     - SENDGRID_API_KEY"
+    echo "  3. Os requisitos de memória estão adequados"
 else
-    echo "✨ Workspace limpo, nenhuma alteração para Hugging Face"
+    echo "❌ Erro durante o deploy para Hugging Face"
+    exit 1
 fi 

src/domain/detectors/cpu.py

@@ -60,7 +60,7 @@ class WeaponDetectorCPU(BaseDetector):
             torch.set_num_interop_threads(min(8, os.cpu_count()))
             
             # Carregar modelo com configurações otimizadas
-            cache_dir = os.getenv('CACHE_DIR', '/tmp/weapon_detection_cache')
+            cache_dir = os.path.join(tempfile.gettempdir(), 'weapon_detection_cache')
             os.makedirs(cache_dir, exist_ok=True)
             
             model_name = "google/owlv2-base-patch16"

src/domain/detectors/gpu.py

@@ -10,6 +10,7 @@ import cv2
 from PIL import Image
 from transformers import Owlv2Processor, Owlv2ForObjectDetection
 from .base import BaseDetector, BaseCache
+import tempfile
 
 logger = logging.getLogger(__name__)
 
@@ -48,7 +49,7 @@ class WeaponDetectorGPU(BaseDetector):
             self.device = self._get_best_device()
             
             # Diretório de cache para o modelo
-            cache_dir = os.getenv('CACHE_DIR', '/tmp/weapon_detection_cache')
+            cache_dir = os.path.join(tempfile.gettempdir(), 'weapon_detection_cache')
             os.makedirs(cache_dir, exist_ok=True)
             
             # Limpar memória GPU