Update chatbot.py

chatbot.py

@@ -27,14 +27,14 @@ import io  # Add this import for working with image bytes
 
 # You can also use models that are commented below
 # model_id = "llava-hf/llava-interleave-qwen-0.5b-hf"
-model_id = "llava-hf/llava-interleave-qwen-7b-hf"
-# model_id = "llava-hf/llava-interleave-qwen-7b-dpo-hf"
+# model_id = "llava-hf/llava-interleave-qwen-7b-hf"
+model_id = "llava-hf/llava-interleave-qwen-7b-dpo-hf"
 processor = LlavaProcessor.from_pretrained(model_id)
 model = LlavaForConditionalGeneration.from_pretrained(model_id, torch_dtype=torch.float16, use_flash_attention_2=True, low_cpu_mem_usage=True)
 model.to("cuda")
 # Credit to merve for code of llava interleave qwen
 
-def sample_frames(video_file, num_frames) :
+def sample_frames(video_file) :
     try:
         video = cv2.VideoCapture(video_file)
         total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))