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 = "llava-v1.5-7b" #TODO 
+llva_model_path = "llava-hf/llava-1.5-7b-hf" #TODO 
 llva_tokenizer, llva_model, llva_image_processor, llva_context_len = load_pretrained_model(
     model_path=llva_model_path,
     model_base=None,
@@ -50,8 +50,8 @@ def Enhance_prompt(prompt,select_images):
 script_directory = os.path.dirname(os.path.realpath(__file__))
 device = "cuda"
 # TODO 
-base_model_path = "Realistic_Vision_V6.0_B1_noVAE" # TODO
-consistentID_path = "ConsistentID" # TODO
+base_model_path = "SG161222/Realistic_Vision_V6.0_B1_noVAE" # TODO
+consistentID_path = "JackAILab/ConsistentID" # TODO
 
 ### Load base model
 pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(