img size fix

app.py

@@ -92,18 +92,18 @@ def set_image_resolution(image, percentage):
     resized_img = cv2.resize(img, (new_width, new_height))
     return resized_img
 
-def predict(link, text, threshold, image_resolution):
+def predict(link, url, threshold, image_resolution):
 
     categories = read_json_categories(os.getcwd() + '/annotations/instances_train2017.json')
 
     if(link):
         img = read_image(link)
     else:
-        img = read_image(text)
-    img = treat_grayscale(img)
-    img = img[:, :, ::-1]
-    print(type(img))
+        img = read_image(url)
+
     img_resized = set_image_resolution(img, image_resolution)
+    img = treat_grayscale(img_resized)
+    img = img[:, :, ::-1]
 
     predictions, visualized_output = demo.run_on_image(img_resized, threshold)