Spaces:
Running
Running
makewong
commited on
Commit
•
045312c
1
Parent(s):
eb2919a
Update pic2song.py
Browse files- pic2song.py +1 -1
pic2song.py
CHANGED
@@ -70,7 +70,7 @@ def predict_step(image_paths):
|
|
70 |
@torch.no_grad()
|
71 |
@torch.inference_mode()
|
72 |
def pic2song(image_paths, image_ins):
|
73 |
-
in_max_new_tokens = int(image_ins *
|
74 |
# 调用predict_step函数并输出预测的图片描述文本列表
|
75 |
print(image_paths)
|
76 |
user_message = predict_step(image_paths)
|
|
|
70 |
@torch.no_grad()
|
71 |
@torch.inference_mode()
|
72 |
def pic2song(image_paths, image_ins):
|
73 |
+
in_max_new_tokens = int(image_ins * 60)
|
74 |
# 调用predict_step函数并输出预测的图片描述文本列表
|
75 |
print(image_paths)
|
76 |
user_message = predict_step(image_paths)
|