test

app.py

@@ -51,8 +51,8 @@ for i, j in zip(ents, ents_prompt):
 
 model_mapping = {
     'gpt3.5': 'gpt-3.5-turbo-0613',
-    'vicuna-7b': 'lmsys/vicuna-7b-v1.3',
-    'llama-7b': './llama/hf/7B',
+    #'vicuna-7b': 'lmsys/vicuna-7b-v1.3',
+    #'llama-7b': './llama/hf/7B',
 }
 
 with open('sample_uniform_1k_2.txt', 'r') as f:
@@ -89,8 +89,8 @@ theme = gr.themes.Soft()
 
 # issue get request for gpt 3.5
 gpt_pipeline = pipeline(task="text2text-generation", model="gpt3.5")
-vicuna7b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-7b-v1.3")
-llama7b_pipeline = pipeline(task="text2text-generation", model="./llama/hf/7B")
+#vicuna7b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-7b-v1.3")
+#llama7b_pipeline = pipeline(task="text2text-generation", model="./llama/hf/7B")
 
 # Dropdown options for model and task
 model_options = list(model_mapping.keys())