Update app.py

app.py

@@ -26,7 +26,7 @@ class Config:
 
 
 # Assuming 'predict_stability' is your function that predicts protein stability
-def predict_stability(cfg, model_choice, organism_choice, pdb_file=None, sequence=None):
+def predict_stability(model_choice, organism_choice, pdb_file=None, sequence=None, cfg=Config()):
     # Check if pdb_file is provided
     if pdb_file:
         pdb_path = pdb_file.name  # Get the path of the uploaded PDB file
@@ -152,7 +152,7 @@ with gr.Blocks() as demo:
 
             predict_button.click(
                 fn=predict_stability,
-                inputs=[Config, model_choice, organism_choice, pdb_file],
+                inputs=[model_choice, organism_choice, pdb_file],
                 outputs=prediction_output,
             )
 
@@ -170,7 +170,7 @@ with gr.Blocks() as demo:
 
             predict_button.click(
                 fn=predict_stability,
-                inputs=[Config, model_choice, organism_choice, sequence],
+                inputs=[model_choice, organism_choice, sequence],
                 outputs=prediction_output,
             )