Marcus Vinicius Zerbini Canhaço commited on
Commit
ba1dd7c
·
1 Parent(s): 619e8a7

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

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. src/main.py +3 -6
requirements.txt CHANGED
@@ -13,7 +13,7 @@ filelock
13
  Flask
14
  fsspec
15
  future
16
- gradio
17
  gradio_client
18
  h11
19
  httpcore
 
13
  Flask
14
  fsspec
15
  future
16
+ gradio==5.15.0
17
  gradio_client
18
  h11
19
  httpcore
src/main.py CHANGED
@@ -40,18 +40,15 @@ def main():
40
 
41
  # Primeiro configurar a fila
42
  demo = demo.queue(
43
- max_size=16, # Aumentado para corresponder ao max_batch_size
44
- concurrency_limit=max_concurrent, # Baseado na memória GPU
45
- status_update_rate=10, # Atualizações mais frequentes
46
  api_open=False,
47
- max_batch_size=16 # Aumentado para corresponder ao detector
48
  )
49
  # Depois fazer o launch
50
  demo.launch(
51
  server_name="0.0.0.0",
52
  server_port=7860,
53
- share=False,
54
- max_threads=4 # Limitar threads da CPU
55
  )
56
  else:
57
  # Ambiente local - apenas launch direto
 
40
 
41
  # Primeiro configurar a fila
42
  demo = demo.queue(
43
+ concurrency_count=1, # Simplificando para um worker
 
 
44
  api_open=False,
45
+ status_update_rate="auto"
46
  )
47
  # Depois fazer o launch
48
  demo.launch(
49
  server_name="0.0.0.0",
50
  server_port=7860,
51
+ share=False
 
52
  )
53
  else:
54
  # Ambiente local - apenas launch direto