Salimshakeel commited on
Commit
891c2e7
·
1 Parent(s): ad3cc6c

fix static path

Browse files
Files changed (3) hide show
  1. config.py +2 -2
  2. main.py +1 -1
  3. routes/summarize.py +1 -1
config.py CHANGED
@@ -1,7 +1,7 @@
1
  # config.py
2
  import torch
3
- UPLOAD_DIR = "backend/static/uploads"
4
- OUTPUT_DIR = "backend/static/outputs"
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 = os.path.join("backend", "static")
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("backend/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}"
 
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}"