mateo496 commited on
Commit
3c9ac96
·
verified ·
1 Parent(s): e5dcb2c

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. .gitattributes +1 -0
  2. server.py +1 -1
.gitattributes ADDED
@@ -0,0 +1 @@
 
 
1
+ *.pt filter=lfs diff=lfs merge=lfs -text
server.py CHANGED
@@ -28,7 +28,7 @@ app.add_middleware(
28
  )
29
 
30
  device = "cuda" if torch.cuda.is_available() else "cpu"
31
- predictor = AudioPredictor("models/cnn/saved/final_model.pt", device=device)
32
 
33
 
34
  @app.get("/")
 
28
  )
29
 
30
  device = "cuda" if torch.cuda.is_available() else "cpu"
31
+ predictor = AudioPredictor("final_model_deployed.pt", device=device)
32
 
33
 
34
  @app.get("/")