reidddd commited on
Commit
a8df9e4
·
1 Parent(s): 93224ed

add fetch image api

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ import io
9
  import os
10
  import requests
11
  import gdown
12
- from skimage import io
13
  from torchvision.ops import box_iou
14
  import torch
15
 
@@ -84,7 +84,7 @@ def upload():
84
  return jsonify({"error": "No file selected"}), 400
85
 
86
  # Load image
87
- image = io.imread(file)
88
  image_np = image
89
 
90
  # Run model prediction
 
9
  import os
10
  import requests
11
  import gdown
12
+ from skimage import io as skio
13
  from torchvision.ops import box_iou
14
  import torch
15
 
 
84
  return jsonify({"error": "No file selected"}), 400
85
 
86
  # Load image
87
+ image = skio.imread(file)
88
  image_np = image
89
 
90
  # Run model prediction