Commit
·
891c2e7
1
Parent(s):
ad3cc6c
fix static path
Browse files- config.py +2 -2
- main.py +1 -1
- routes/summarize.py +1 -1
config.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
# config.py
|
2 |
import torch
|
3 |
-
UPLOAD_DIR = "
|
4 |
-
OUTPUT_DIR = "
|
5 |
FRAME_RATE = 15
|
6 |
SCORE_THRESHOLD = 0.4
|
7 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
1 |
# config.py
|
2 |
import torch
|
3 |
+
UPLOAD_DIR = "static/uploads"
|
4 |
+
OUTPUT_DIR = "static/outputs"
|
5 |
FRAME_RATE = 15
|
6 |
SCORE_THRESHOLD = 0.4
|
7 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
main.py
CHANGED
@@ -22,5 +22,5 @@ app.add_middleware(
|
|
22 |
)
|
23 |
|
24 |
# Mount static folder
|
25 |
-
static_dir =
|
26 |
app.mount("/static", StaticFiles(directory=static_dir), name="static")
|
|
|
22 |
)
|
23 |
|
24 |
# Mount static folder
|
25 |
+
static_dir = "static"
|
26 |
app.mount("/static", StaticFiles(directory=static_dir), name="static")
|
routes/summarize.py
CHANGED
@@ -15,7 +15,7 @@ def summarize_video(video: UploadFile = File(...)):
|
|
15 |
|
16 |
video_path = save_uploaded_file(video, UPLOAD_DIR)
|
17 |
features, picks = extract_features(video_path)
|
18 |
-
model = load_model("
|
19 |
scores = get_scores(model, features)
|
20 |
selected = get_selected_indices(scores, picks)
|
21 |
output_path = f"{OUTPUT_DIR}/summary_{video.filename}"
|
|
|
15 |
|
16 |
video_path = save_uploaded_file(video, UPLOAD_DIR)
|
17 |
features, picks = extract_features(video_path)
|
18 |
+
model = load_model("Model/epoch-199.pkl")
|
19 |
scores = get_scores(model, features)
|
20 |
selected = get_selected_indices(scores, picks)
|
21 |
output_path = f"{OUTPUT_DIR}/summary_{video.filename}"
|