Zeph27 commited on
Commit
b04464a
Β·
1 Parent(s): 5bf65b0
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from PIL import Image
5
  from decord import VideoReader, cpu
6
  import base64
7
  import io
 
8
 
9
  # Load model
10
  model_path = 'openbmb/MiniCPM-V-2_6'
@@ -41,6 +42,7 @@ def encode_video(video_path):
41
  video = [encode_image(v) for v in video]
42
  return video
43
 
 
44
  def analyze_video(prompt, video):
45
  encoded_video = encode_video(video.name)
46
 
 
5
  from decord import VideoReader, cpu
6
  import base64
7
  import io
8
+ import spaces
9
 
10
  # Load model
11
  model_path = 'openbmb/MiniCPM-V-2_6'
 
42
  video = [encode_image(v) for v in video]
43
  return video
44
 
45
+ @spaces.GPU
46
  def analyze_video(prompt, video):
47
  encoded_video = encode_video(video.name)
48