Spaces:
Runtime error
Runtime error
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(
|