Update app.py

app.py

@@ -15,7 +15,7 @@ from llava.mm_utils import get_model_name_from_path
 from llava.eval.run_llava import eval_model
 
 # Load Lava for prompt enhancement
-llva_model_path = "/data6/huangjiehui_m22/pretrained_model/llava-v1.5-7b" #TODO 
+llva_model_path = "llava-v1.5-7b" #TODO 
 llva_tokenizer, llva_model, llva_image_processor, llva_context_len = load_pretrained_model(
     model_path=llva_model_path,
     model_base=None,
@@ -45,13 +45,13 @@ def Enhance_prompt(prompt,select_images):
 
 # print(gr.__version__)
 # 4.16.0
-os.environ['GRADIO_TEMP_DIR'] = "/data6/huangjiehui_m22/z_benke/liaost/ConsistentID/images/gradio_tmp" #TODO
+
 
 script_directory = os.path.dirname(os.path.realpath(__file__))
 device = "cuda"
 # TODO 
-base_model_path = "/data6/huangjiehui_m22/pretrained_model/Realistic_Vision_V6.0_B1_noVAE" # TODO
-consistentID_path = "/data6/huangjiehui_m22/z_benke/liaost/ConsistentID/models/ConsistentID_model_facemask_pretrain_50w.bin" # TODO
+base_model_path = "Realistic_Vision_V6.0_B1_noVAE" # TODO
+consistentID_path = "ConsistentID" # TODO
 
 ### Load base model
 pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(