XiaoyiYangRIT commited on
Commit
e041131
·
1 Parent(s): aed9794

Update some files

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ def extract_frames(video_path, num_frames=8):
91
 
92
  # === 主推理函数 ===
93
  def evaluate_ar(video):
94
- frames = extract_frames(video.name).to(torch.bfloat16).cuda()
95
  prompt = "Evaluate the quality of AR occlusion and rendering in the uploaded video."
96
  num_patches = [1] * frames.shape[0]
97
  output, _ = model.chat(
 
91
 
92
  # === 主推理函数 ===
93
  def evaluate_ar(video):
94
+ frames = extract_frames(video).to(torch.bfloat16).cuda()
95
  prompt = "Evaluate the quality of AR occlusion and rendering in the uploaded video."
96
  num_patches = [1] * frames.shape[0]
97
  output, _ = model.chat(