Update app.py
Browse files
app.py
CHANGED
@@ -94,7 +94,7 @@ def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
|
|
94 |
|
95 |
# Salve o resultado em um arquivo se a opção estiver ativada
|
96 |
if salvar_resultado:
|
97 |
-
nome_arquivo = f'resultado_{int(time.time())}.png'
|
98 |
cv2.imwrite(nome_arquivo, img_out)
|
99 |
|
100 |
return spent, img_out
|
@@ -104,6 +104,7 @@ def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
|
|
104 |
return str(e), None
|
105 |
|
106 |
|
|
|
107 |
# Carregar modelos pré-treinados
|
108 |
models = {
|
109 |
'SE-RegUNet 4GF': torch.jit.load('./model/SERegUNet4GF.pt'),
|
|
|
94 |
|
95 |
# Salve o resultado em um arquivo se a opção estiver ativada
|
96 |
if salvar_resultado:
|
97 |
+
nome_arquivo = f'Salvar Resultado/resultado_{int(time.time())}.png'
|
98 |
cv2.imwrite(nome_arquivo, img_out)
|
99 |
|
100 |
return spent, img_out
|
|
|
104 |
return str(e), None
|
105 |
|
106 |
|
107 |
+
|
108 |
# Carregar modelos pré-treinados
|
109 |
models = {
|
110 |
'SE-RegUNet 4GF': torch.jit.load('./model/SERegUNet4GF.pt'),
|