Updated

app.py

@@ -30,7 +30,8 @@ from scipy.special import expit
 
 # Load model and move to device
 #checkpoint = 'ThorbenF/prot_t5_xl_uniref50'
-checkpoint = 'ThorbenF/prot_t5_xl_uniref50_cryptic'
+#checkpoint = 'ThorbenF/prot_t5_xl_uniref50_cryptic'
+checkpoint = 'ThorbenF/prot_t5_xl_uniref50_database'
 max_length = 1500
 model, tokenizer = load_model(checkpoint, max_length)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -532,4 +533,4 @@ with gr.Blocks(css="""
         outputs=[predictions_output, molecule_output, download_output]
     )
 
-demo.launch(share=True)
+demo.launch(share=True)