Julia-Amadio
commited on
Commit
·
667a199
1
Parent(s):
0f0e135
Voltando linha crucial que exclui.
Browse files
main.py
CHANGED
|
@@ -47,12 +47,13 @@ async def extract_features(file: UploadFile = File(...), token: str = Depends(ve
|
|
| 47 |
"""
|
| 48 |
#A única pasta em um contêiner Docker que tem permissão de escrita garantida é a pasta /tmp
|
| 49 |
#Precisamos direcionar todos os arquivos temporários para lá
|
|
|
|
|
|
|
| 50 |
with open(temp_path, "wb") as buffer:
|
| 51 |
shutil.copyfileobj(file.file, buffer)
|
| 52 |
|
| 53 |
-
#Processa a imagem (agora retorna o array de dados)
|
| 54 |
features_array = process_single_image(temp_path, output_dir="/tmp")
|
| 55 |
-
|
| 56 |
#Remove imagem temporária
|
| 57 |
os.remove(temp_path)
|
| 58 |
|
|
|
|
| 47 |
"""
|
| 48 |
#A única pasta em um contêiner Docker que tem permissão de escrita garantida é a pasta /tmp
|
| 49 |
#Precisamos direcionar todos os arquivos temporários para lá
|
| 50 |
+
temp_path = f"/tmp/temp_{file.filename}"
|
| 51 |
+
|
| 52 |
with open(temp_path, "wb") as buffer:
|
| 53 |
shutil.copyfileobj(file.file, buffer)
|
| 54 |
|
|
|
|
| 55 |
features_array = process_single_image(temp_path, output_dir="/tmp")
|
| 56 |
+
|
| 57 |
#Remove imagem temporária
|
| 58 |
os.remove(temp_path)
|
| 59 |
|