imperiusrex commited on
Commit
0a834b8
·
verified ·
1 Parent(s): fe05f3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,7 +7,7 @@ import spaces
7
  from ultralytics import YOLO
8
  from tqdm import tqdm
9
  from PIL import Image
10
- from transformers import LlavaNextProcessor, LlavaNextForConditionalGeneration
11
 
12
  # Prevent config warnings
13
  os.environ["YOLO_CONFIG_DIR"] = "/tmp"
@@ -90,8 +90,7 @@ def process_video(video_path):
90
  sharp = cv2.addWeighted(selective,2.0,blur,-1.0,0)
91
  out_img = "sharpened_board_color.jpg"
92
  cv2.imwrite(out_img, sharp)
93
- caption = caption_image_with_llava(out_img)
94
- return out_img, caption
95
 
96
  demo = gr.Interface(
97
  fn=process_video,
 
7
  from ultralytics import YOLO
8
  from tqdm import tqdm
9
  from PIL import Image
10
+
11
 
12
  # Prevent config warnings
13
  os.environ["YOLO_CONFIG_DIR"] = "/tmp"
 
90
  sharp = cv2.addWeighted(selective,2.0,blur,-1.0,0)
91
  out_img = "sharpened_board_color.jpg"
92
  cv2.imwrite(out_img, sharp)
93
+
 
94
 
95
  demo = gr.Interface(
96
  fn=process_video,