Freeze numpy version
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -80,7 +80,7 @@ def main(video, query, model, cfg):
|
|
| 80 |
mr = [[convert_time(p[0]), convert_time(p[1]), round(p[2], 2)] for p in mr]
|
| 81 |
|
| 82 |
hd = pred['_out']['saliency'].cpu()
|
| 83 |
-
hd = ((hd - hd.min()) / (hd.max() - hd.min())).tolist()
|
| 84 |
hd = pd.DataFrame(dict(x=range(0, len(hd) * 2, 2), y=hd))
|
| 85 |
|
| 86 |
return mr, hd
|
|
|
|
| 80 |
mr = [[convert_time(p[0]), convert_time(p[1]), round(p[2], 2)] for p in mr]
|
| 81 |
|
| 82 |
hd = pred['_out']['saliency'].cpu()
|
| 83 |
+
hd = ((hd - hd.min()) / (hd.max() - hd.min()) * 0.9 + 0.05).tolist()
|
| 84 |
hd = pd.DataFrame(dict(x=range(0, len(hd) * 2, 2), y=hd))
|
| 85 |
|
| 86 |
return mr, hd
|
requirements.txt
CHANGED
|
@@ -2,6 +2,7 @@ git+https://github.com/openai/CLIP.git@a1d0717
|
|
| 2 |
decord==0.6.0
|
| 3 |
gradio==4.41.0
|
| 4 |
nncore==0.4.4
|
|
|
|
| 5 |
pandas==2.2.2
|
| 6 |
torch==2.2.1
|
| 7 |
torchvision==0.17.1
|
|
|
|
| 2 |
decord==0.6.0
|
| 3 |
gradio==4.41.0
|
| 4 |
nncore==0.4.4
|
| 5 |
+
numpy==1.26.4
|
| 6 |
pandas==2.2.2
|
| 7 |
torch==2.2.1
|
| 8 |
torchvision==0.17.1
|