Bhanushray commited on
Commit
a16c4b0
Β·
verified Β·
1 Parent(s): 84f44e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,6 +17,7 @@ import re
17
  bio_model_dir = "/app/modelsBioembedSmall" # Persistent model storage
18
  cvn_model_dir = "/app/models_folder"
19
  UPLOAD_FOLDER = "/app/Samples"
 
20
 
21
  os.makedirs(bio_model_dir, exist_ok=True)
22
  os.makedirs(cvn_model_dir, exist_ok=True)
@@ -194,7 +195,7 @@ def generate_smiles(sequence, n_samples=100):
194
  if not smiles_list:
195
  return None, "No valid SMILES generated!"
196
 
197
- filename = os.path.join(UPLOAD_FOLDER, "SMILES_GENERATED.txt")
198
  with open(filename, "w") as file:
199
  file.write("\n".join(smiles_list))
200
 
@@ -221,7 +222,7 @@ def index():
221
 
222
  @app.route("/download")
223
  def download_file():
224
- file_path = os.path.join(UPLOAD_FOLDER, "SMILES_GENERATED.txt")
225
  return send_file(file_path, as_attachment=True)
226
 
227
  # πŸš€ Run the Flask App on Railway
 
17
  bio_model_dir = "/app/modelsBioembedSmall" # Persistent model storage
18
  cvn_model_dir = "/app/models_folder"
19
  UPLOAD_FOLDER = "/app/Samples"
20
+ UF="/tmp/"
21
 
22
  os.makedirs(bio_model_dir, exist_ok=True)
23
  os.makedirs(cvn_model_dir, exist_ok=True)
 
195
  if not smiles_list:
196
  return None, "No valid SMILES generated!"
197
 
198
+ filename = os.path.join(UF, "SMILES_GENERATED.txt")
199
  with open(filename, "w") as file:
200
  file.write("\n".join(smiles_list))
201
 
 
222
 
223
  @app.route("/download")
224
  def download_file():
225
+ file_path = os.path.join(UF, "SMILES_GENERATED.txt")
226
  return send_file(file_path, as_attachment=True)
227
 
228
  # πŸš€ Run the Flask App on Railway