Update app.py

app.py

@@ -6,7 +6,7 @@ st.set_page_config(page_title="NeoProtein Designer", page_icon="🧬")
 st.title("🧬 NeoProtein-GPT Protein Designer")
 st.markdown("""
 ### Design novel protein sequences with unique binding sites
-*Using the [NeoProtein-GPT](https://huggingface.co/ayyuce/NeoProtein-GPT) model*
+*Using the [NeoProtein-GPT](https://huggingface.co/MLforHealthcare/NeoProtein-GPT) model*
 """)
 
 with st.sidebar:
@@ -19,13 +19,13 @@ with st.sidebar:
 @st.cache_resource(show_spinner=False)
 def load_model():
     model = GPT2LMHeadModel.from_pretrained(
-        "ayyuce/NeoProtein-GPT",
+        "MLforHealthcare/NeoProtein-GPT",
         force_download=True,
         resume_download=False,
         local_files_only=False,
         trust_remote_code=True
     )
-    tokenizer = GPT2Tokenizer.from_pretrained("ayyuce/NeoProtein-GPT")
+    tokenizer = GPT2Tokenizer.from_pretrained("MLforHealthcare/NeoProtein-GPT")
     return model, tokenizer
 
 model, tokenizer = load_model()