Spaces:
Sleeping
Sleeping
XiaoyiYangRIT
commited on
Commit
·
e041131
1
Parent(s):
aed9794
Update some files
Browse files
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
|
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(
|