Spaces:
Running
Running
Enzo Reis de Oliveira
commited on
Commit
·
3fc57b3
1
Parent(s):
18a87d1
Changing the name csv output
Browse files
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 |
-
|
39 |
-
df.to_csv(
|
40 |
-
|
41 |
-
|
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)
|