Update app.py

app.py

@@ -7,7 +7,8 @@ from huggingface_hub import hf_hub_download
 model_path = hf_hub_download(repo_id="SalmanAboAraj/Tooth1", filename="unet_model.h5") 
 model = load_model(model_path) 
 def predict(image): 
-    original_height, original_width, _ = image.shape image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) 
+    original_height, original_width, _ = image.shape 
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) 
     image = cv2.resize(image, (128, 128)) 
     image = np.expand_dims(image, axis=0) 
     image = np.expand_dims(image, axis=-1)