Julia-Amadio commited on
Commit
667a199
·
1 Parent(s): 0f0e135

Voltando linha crucial que exclui.

Browse files
Files changed (1) hide show
  1. main.py +3 -2
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