Update app.py

app.py

@@ -15,12 +15,25 @@ from matplotlib import pyplot as plt
 from torchvision import transforms
 
 # init the model
+#args = {
+#    'model': 'openllama_peft',
+#    'imagebind_ckpt_path': './pretrained_ckpt/imagebind_ckpt/imagebind_huge.pth',
+#    'vicuna_ckpt_path': './pretrained_ckpt/vicuna_ckpt/7b_v0',
+#    'anomalygpt_ckpt_path': './ckpt/train_supervised/pytorch_model.pt',
+#    'delta_ckpt_path': './pretrained_ckpt/pandagpt_ckpt/7b/pytorch_model.pt',
+#    'stage': 2,
+#    'max_tgt_len': 128,
+#    'lora_r': 32,
+#    'lora_alpha': 32,
+#    'lora_dropout': 0.1
+#}
+
 args = {
     'model': 'openllama_peft',
-    'imagebind_ckpt_path': './pretrained_ckpt/imagebind_ckpt/imagebind_huge.pth',
-    'vicuna_ckpt_path': './pretrained_ckpt/vicuna_ckpt/7b_v0',
-    'anomalygpt_ckpt_path': './ckpt/train_supervised/pytorch_model.pt',
-    'delta_ckpt_path': './pretrained_ckpt/pandagpt_ckpt/7b/pytorch_model.pt',
+    'imagebind_ckpt_path': 'seung275/imagebind_huge.pth',
+    'vicuna_ckpt_path': 'seung275/7b_v0',
+    'anomalygpt_ckpt_path': '/seung275/anomalygpt_ckpt_path',
+    'delta_ckpt_path': '/seung275/delta_ckpt_path',
     'stage': 2,
     'max_tgt_len': 128,
     'lora_r': 32,
@@ -28,6 +41,7 @@ args = {
     'lora_dropout': 0.1
 }
 
+
 model = OpenLLAMAPEFTModel(**args)
 delta_ckpt = torch.load(args['delta_ckpt_path'], map_location=torch.device('cpu'))
 model.load_state_dict(delta_ckpt, strict=False)