Update testforhuggingface.py

testforhuggingface.py

@@ -8,7 +8,7 @@ Original file is located at
 """
 
 # Commented out IPython magic to ensure Python compatibility.
-# %cd drive/MyDrive/protein-bert/proteinbert/
+
 
 
 !pip install git+'https://github.com/miladansari/protein_bert.git'
@@ -29,8 +29,8 @@ from pandas import read_csv
 
 #reading files.
 
-test = read_csv('/content/drive/MyDrive/protein-bert/BENCHMARK_DIR/hemolytic.test.csv',skipinitialspace=True)
-train = read_csv('/content/drive/MyDrive/protein-bert/BENCHMARK_DIR/hemolytic.train.csv',skipinitialspace=True)
+test = read_csv('hemolytic.test.csv',skipinitialspace=True)
+train = read_csv('hemolytic.train.csv',skipinitialspace=True)
 
 seqs=train['seq']
 seqs_test=test['seq']
@@ -44,7 +44,7 @@ from proteinbert import load_pretrained_model
 from proteinbert.conv_and_global_attention_model import get_model_with_hidden_layers_as_outputs
 
 #load pretrained model.
-pretrained_model_generator, input_encoder = load_pretrained_model(local_model_dump_dir='/content/drive/MyDrive/protein-bert/proteinbert')
+pretrained_model_generator, input_encoder = load_pretrained_model(local_model_dump_dir='proteinbert_preModel')
 model = get_model_with_hidden_layers_as_outputs(pretrained_model_generator.create_model(seq_len))
 
 #extract local representaion (Embeddings).