Boboiazumi commited on
Commit
c20c7fc
1 Parent(s): 6422051

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ def load_pipeline(model_name):
84
  img_pipe.to(device)
85
  return pipe, img_pipe
86
 
87
- def load_img(resize_width,img):
88
  img = Image.open(img)
89
  width, height = img.size
90
  scale = resize_width / width
 
84
  img_pipe.to(device)
85
  return pipe, img_pipe
86
 
87
+ def load_img(resize_width: float,img: str):
88
  img = Image.open(img)
89
  width, height = img.size
90
  scale = resize_width / width