Update app.py

app.py

@@ -45,18 +45,18 @@ bise_net.cuda()
 
 import sys
 sys.path.append("./models/LLaVA")
-from llava.model.builder import load_pretrained_model
-from llava.mm_utils import get_model_name_from_path
-from llava.eval.run_llava import eval_model
+# from llava.model.builder import load_pretrained_model
+# from llava.mm_utils import get_model_name_from_path
+# from llava.eval.run_llava import eval_model
 
 # Load Llava for prompt enhancement
 llva_model_path = "liuhaotian/llava-v1.5-7b"  
-llva_tokenizer, llva_model, llva_image_processor, llva_context_len = load_pretrained_model(
-    model_path=llva_model_path,
-    model_base=None,
-    model_name=get_model_name_from_path(llva_model_path),)    
+# llva_tokenizer, llva_model, llva_image_processor, llva_context_len = load_pretrained_model(
+#     model_path=llva_model_path,
+#     model_base=None,
+#     model_name=get_model_name_from_path(llva_model_path),)    
 # llva_tokenizer.cuda()
-llva_model.to(device)
+# llva_model.to(device)
 # llva_image_processor.to(device)
 
 ### Load consistentID_model checkpoint
@@ -102,20 +102,20 @@ def process(selected_template_images,costum_image,prompt
     def Enhance_prompt(prompt,select_images):
         
         llva_prompt = f'Please ignore the image. Enhance the following text prompt for me. You can associate more details with the character\'s gesture, environment, and decent clothing:"{prompt}".' 
-        args = type('Args', (), {
-            "model_path": llva_model_path,
-            "model_base": None,
-            "model_name": get_model_name_from_path(llva_model_path),
-            "query": llva_prompt,
-            "conv_mode": None,
-            "image_file": select_images,
-            "sep": ",",
-            "temperature": 0,
-            "top_p": None,
-            "num_beams": 1,
-            "max_new_tokens": 512
-        })() 
-        Enhanced_prompt = eval_model(args, llva_tokenizer, llva_model, llva_image_processor)
+        # args = type('Args', (), {
+        #     "model_path": llva_model_path,
+        #     "model_base": None,
+        #     "model_name": get_model_name_from_path(llva_model_path),
+        #     "query": llva_prompt,
+        #     "conv_mode": None,
+        #     "image_file": select_images,
+        #     "sep": ",",
+        #     "temperature": 0,
+        #     "top_p": None,
+        #     "num_beams": 1,
+        #     "max_new_tokens": 512
+        # })() 
+        # Enhanced_prompt = eval_model(args, llva_tokenizer, llva_model, llva_image_processor)
     
         return Enhanced_prompt