Update app.py
Browse files
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("
|
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
|