Update app.py

app.py

@@ -14,6 +14,7 @@ from mplug_owl_video.processing_mplug_owl import MplugOwlImageProcessor, MplugOw
 from peft import LoraConfig, get_peft_model
 from data_utils.xgpt3_dataset import MultiModalDataset
 from utils import batchify
+from huggingface_hub import hf_hub_download
 
 import gradio as gr
 from entailment_inference import get_scores
@@ -35,8 +36,9 @@ def modify_keys(state_dict):
 
     return new_state_dict
 
-pretrained_ckpt = "mplugowl7bvideo/"
-trained_ckpt = "owl-con/checkpoint-5178/pytorch_model.bin"
+pretrained_ckpt = "MAGAer13/mplug-owl-llama-7b-video/"
+trained_ckpt = hf_hub_download(repo_id="videocon/owl-con", filename="pytorch_model.bin", repo_type="model")
+
 
 tokenizer = LlamaTokenizer.from_pretrained(pretrained_ckpt)
 image_processor = MplugOwlImageProcessor.from_pretrained(pretrained_ckpt)