Update app.py

app.py

@@ -106,8 +106,8 @@ try:
     # Using a smaller model that is only ~200 MB
     model_name = "facebook/esm2_t6_8M_UR50D"  # Smaller model with 320-dim embeddings
 
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModel.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(bio_model_dir)
+    model = AutoModel.from_pretrained(bio_model_dir)
     model.eval()
     print("ESM2 model loaded.")
 except Exception as e: