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- requirements.txt +1 -1
- 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 |
-
|
44 |
-
concurrency_limit=max_concurrent, # Baseado na memória GPU
|
45 |
-
status_update_rate=10, # Atualizações mais frequentes
|
46 |
api_open=False,
|
47 |
-
|
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
|