Update app.py
Browse files
app.py
CHANGED
@@ -32,26 +32,27 @@ def predict_stability(cfg, model_choice, organism_choice, pdb_file=None, sequenc
|
|
32 |
pdb_path = pdb_file.name # Get the path of the uploaded PDB file
|
33 |
os.system("chmod 777 bin/foldseek")
|
34 |
sequences = get_foldseek_seq(pdb_path)
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
#
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
|
|
55 |
|
56 |
|
57 |
def get_foldseek_seq(pdb_path):
|
|
|
32 |
pdb_path = pdb_file.name # Get the path of the uploaded PDB file
|
33 |
os.system("chmod 777 bin/foldseek")
|
34 |
sequences = get_foldseek_seq(pdb_path)
|
35 |
+
return str(sequences)
|
36 |
+
# if not sequences:
|
37 |
+
# return "Failed to extract sequence from the PDB file."
|
38 |
+
# if model_choice == "SaProt":
|
39 |
+
# sequence = sequences[2]
|
40 |
+
# else:
|
41 |
+
# sequence = sequences[0]
|
42 |
+
|
43 |
+
# if organism_choice == "Human":
|
44 |
+
# cell_line = "HeLa"
|
45 |
+
# else:
|
46 |
+
# cell_line = "NIH3T3"
|
47 |
+
# # If sequence is provided directly
|
48 |
+
# if sequence:
|
49 |
+
# cfg.model = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
|
50 |
+
# cfg.architecture = model_choice
|
51 |
+
# cfg.model_path = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
|
52 |
+
# output = predict(cfg, sequence)
|
53 |
+
# return f"Predicted Stability using {model_choice} for {organism_choice}: Example Output with sequence {output}..."
|
54 |
+
# else:
|
55 |
+
# return "No valid input provided."
|
56 |
|
57 |
|
58 |
def get_foldseek_seq(pdb_path):
|