Update app.py

app.py

@@ -44,15 +44,14 @@ def predict_stability(model_choice, organism_choice, pdb_file=None, sequence=Non
     else:
         cell_line = "NIH3T3"
     # If sequence is provided directly
-    return str(sequences)
-    # if sequence:
-    #     cfg.model = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
-    #     cfg.architecture = model_choice
-    #     cfg.model_path = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
-    #     output = predict(cfg, sequence)
-    #     return f"Predicted Stability using {model_choice} for {organism_choice}: Example Output with sequence {output}..."
-    # else:
-    #     return "No valid input provided."
+    if sequence:
+        cfg.model = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
+        cfg.architecture = model_choice
+        cfg.model_path = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
+        output = predict(cfg, sequence)
+        return f"Predicted Stability using {model_choice} for {organism_choice}: Example Output with sequence {output}..."
+    else:
+        return "No valid input provided."
 
 
 def get_foldseek_seq(pdb_path):
@@ -73,7 +72,7 @@ def predict(cfg, sequence):
         cfg.max_length += 1
 
     seed_everything(cfg.seed)
-
+    error
     df = pd.DataFrame({cfg.sequence_col: [sequence]})
 
     tokenizer = AutoTokenizer.from_pretrained(