Spaces:

Bhanushray
/

WebAppPTS

Sleeping

Bhanushray commited on Apr 23

Commit

a49f427

verified ·

1 Parent(s): 119c8a6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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)
     model.eval()
     print("ESM2 model loaded.")
 except Exception as e:

     # 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(bio_model_dir)
+    model = AutoModel.from_pretrained(bio_model_dir)
     model.eval()
     print("ESM2 model loaded.")
 except Exception as e: