Spaces:
Sleeping
Sleeping
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)
|