ftx7go commited on
Commit
fec6caf
·
verified ·
1 Parent(s): 3d29769

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from PIL import Image
14
  os.environ["CUDA_VISIBLE_DEVICES"] = ""
15
 
16
  # Load the trained fracture detection model
17
- model = torch.load("model.pth")
18
  model.eval()
19
 
20
  # Function to predict fracture
 
14
  os.environ["CUDA_VISIBLE_DEVICES"] = ""
15
 
16
  # Load the trained fracture detection model
17
+ model = torch.load("my_keras_model.h5")
18
  model.eval()
19
 
20
  # Function to predict fracture