Update app.py

app.py

@@ -6,8 +6,8 @@ import numpy as np
 import pandas as pd
 
 # Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("TianlaiChen/PepMLM-650M")
-model = AutoModelForMaskedLM.from_pretrained("TianlaiChen/PepMLM-650M")
+tokenizer = AutoTokenizer.from_pretrained("ChatterjeeLab/PepMLM-650M")
+model = AutoModelForMaskedLM.from_pretrained("ChatterjeeLab/PepMLM-650M")
 
 def compute_pseudo_perplexity(model, tokenizer, protein_seq, binder_seq):
     sequence = protein_seq + binder_seq