Enzo Reis de Oliveira commited on
Commit
3fc57b3
·
1 Parent(s): 18a87d1

Changing the name csv output

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -33,13 +33,12 @@ def gerar_embedding_e_csv(smiles: str):
33
  try:
34
  # Gera o vetor
35
  vetor = model.encode(smiles, return_torch=True)[0].tolist()
36
- # Grava CSV
37
  df = pd.DataFrame([vetor])
38
- tmp = tempfile.NamedTemporaryFile(suffix=".csv", delete=False)
39
- df.to_csv(tmp.name, index=False)
40
- tmp.close()
41
- # Retorna JSON em string e ativa o link de download
42
- return json.dumps(vetor), gr.update(value=tmp.name, visible=True)
43
  except Exception as e:
44
  erro = {"erro": str(e)}
45
  return json.dumps(erro), gr.update(visible=False)
 
33
  try:
34
  # Gera o vetor
35
  vetor = model.encode(smiles, return_torch=True)[0].tolist()
36
+ # Grava CSV com nome fixo
37
  df = pd.DataFrame([vetor])
38
+ filename = "embeddings.csv"
39
+ df.to_csv(filename, index=False)
40
+ # Retorna JSON em string e ativa o link de download para embeddings.csv
41
+ return json.dumps(vetor), gr.update(value=filename, visible=True)
 
42
  except Exception as e:
43
  erro = {"erro": str(e)}
44
  return json.dumps(erro), gr.update(visible=False)