Update app.py

app.py

@@ -75,6 +75,7 @@ class model:
  
  return T.ToPILImage()(protein_image), T.ToPILImage()(nucleus_image), formatted_predicted_sequence
 
+base_class = model()
 
 with gr.Blocks(theme='gradio/soft') as demo:
  gr.Markdown("Select the prediction model.")
@@ -143,6 +144,6 @@ with gr.Blocks(theme='gradio/soft') as demo:
 
  outputs = [mask, nucleus_crop, predicted_sequence]
 
- button.click(gradio_demo, inputs, outputs)
+ button.click(base_class.gradio_demo, inputs, outputs)
 
 demo.launch(enable_queue=True)