kadirnar commited on
Commit
1062ed1
1 Parent(s): f269a3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def sahi_yolov5_inference(
98
  return results.render()[0]
99
 
100
  elif model_type == "Unet-Istanbul":
101
- import unet_prediction
102
 
103
  output = unet_prediction(input_path=image, model_path=model_id)
104
  return output
 
98
  return results.render()[0]
99
 
100
  elif model_type == "Unet-Istanbul":
101
+ from ististanbul_unet import unet_prediction
102
 
103
  output = unet_prediction(input_path=image, model_path=model_id)
104
  return output