Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import matplotlib.pyplot as plt
 
 # Load ESM2 model and tokenizer
-MODEL_NAME = "facebook/esm2_t6_8M_UR50D"  # smaller model for Hugging Face Space
+MODEL_NAME = "facebook/esm2_t36_3B_UR50D"  # smaller model for Hugging Face Space
 model = AutoModel.from_pretrained(MODEL_NAME, output_hidden_states=True)
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model.eval()