Update app.py

app.py

@@ -51,7 +51,7 @@ def predict_stability(model_choice, organism_choice, pdb_file=None, sequence=Non
         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}..."
+        return output
     else:
         return "No valid input provided."
 
@@ -114,7 +114,7 @@ def predict(cfg, sequence):
 
     html_output = f"""
     <div style='border: 2px solid #4CAF50; padding: 10px; border-radius: 10px;'>
-        <h3 style='color: #4CAF50;'>Protein: {cfg.sequence_col}</h3>
+        <h3 style='color: #4CAF50;'>Protein: {sequence}</h3>
         <p><strong>Stability:</strong> {outputs['raw prediction values']}</p>
         <p><strong>Organism:</strong> {outputs['binary prediction values']}</p>
     </div>