Update app.py

app.py

@@ -11,11 +11,9 @@ from llava.model.builder import load_pretrained_model
 from llava_utils import prompt_wrapper, generator
 from utils import normalize, denormalize, load_image
 
-BASE_DIR = "/workingdir/soh337/llavaguard"
-# BASE_DIR = "/Users/ahren/Workspace/Multimodal/llavaguard"
 UNCONSTRAINED_ATTACK_IMAGE_PATH = 'unconstrained_attack_images/adversarial_'
 CONSTRAINED_ATTACK_IMAGE_PATH = 'adversarial_qna_images/adv_image_'
-MODEL_PATH = "/workingdir/models_hf/liuhaotian/llava-v1.5-13b"
+MODEL_PATH = "liuhaotian/llava-v1.5-13b"
 
 TEXT_SAFETY_PATCHES = {
     "optimized": "text_patch_optimized",
@@ -59,7 +57,7 @@ print('>>> Initializing Models')
 load_model_async(MODEL_PATH, get_model_name_from_path(MODEL_PATH))
 
 
-prompts = rtp_read(osp.join(BASE_DIR, 'harmful_corpus/rtp_prompts.jsonl'))
+prompts = rtp_read(osp.join('harmful_corpus/rtp_prompts.jsonl'))
 
 # out_unprotected: responses without the safety patch
 out, out_unprotected = [], []