Update app.py

app.py

@@ -49,21 +49,22 @@ for filename in filenames:
         local_dir="./model/"
     )
 
+for items in os.listdir("./model"):
+    print(items)
+
 # def no_logger():
 #     logging.config.dictConfig({
 #         'version': 1,
 #         'disable_existing_loggers': True,
 #     })
 
-MODEL_PATH = os.path.abspath("model")
-print(MODEL_PATH)
 
 model = AutoModelForCausalLM.from_pretrained(
-    MODEL_PATH ,
+    pretrained_model_name_or_path="./model/",
     torch_dtype=torch.float16,
     low_cpu_mem_usage=True,
 ).to(0)
-tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH )
+tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path="./model/")
 vision_tower = model.get_vision_tower()
 vision_tower.load_model()
 vision_tower.to(device="cuda", dtype=torch.float16)