Upload code

modeling_prot2text.py

@@ -1,7 +1,7 @@
 from transformers import GPT2Config, AutoTokenizer, GPT2Config
 from transformers import PretrainedConfig, PreTrainedModel
 import transformers
-from typing import Optional, Tuple, Callable
+from typing import Optional, Tuple, Callable, List
 import torch
 import torch.nn as nn
 from transformers.modeling_utils import PreTrainedModel, PretrainedConfig