only gpt2

app.py

@@ -51,14 +51,6 @@ for i, j in zip(ents, ents_prompt):
 
 model_mapping = {
     'gpt3.5': 'gpt2',
-    'vicuna-7b': 'lmsys/vicuna-7b-v1.3',
-    'vicuna-13b': 'lmsys/vicuna-13b-v1.3',
-    'vicuna-33b': 'lmsys/vicuna-33b-v1.3',
-    'fastchat-t5': 'lmsys/fastchat-t5-3b-v1.0',
-    'llama-7b': './llama/hf/7B',
-    'llama-13b': './llama/hf/13B',
-    'llama-30b': './llama/hf/30B',
-    'alpaca': './alpaca-7B',
 }
 
 with open('sample_uniform_1k_2.txt', 'r') as f: