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
|