Cancer Detector AppV1

app.py

@@ -69,19 +69,19 @@ show=True
 #path = glob.glob('*.png')
 
 def show_preds_image(path):
-    # for image in path:
-    #     img = Image.open(image)
-    #     if show:
-    #         plt.imshow(img)
-    #         plt.show()
-    img = cv2.imread(path)
-    class_name = ['adenocarcinoma',
-                'large.cell.carcinoma',
-                'normal',
-                'squamous.cell.carcinoma']
-    prob, result = predict(img, model, 'cpu', class_name)
-    if display_prob:
-        print('Probability of {} : {:.6f}'.format(result, prob))
+    for image in path:
+        img = Image.open(image)
+        if show:
+            plt.imshow(img)
+            plt.show()
+    #img = cv2.imread(path)
+        class_name = ['adenocarcinoma',
+                    'large.cell.carcinoma',
+                    'normal',
+                    'squamous.cell.carcinoma']
+        prob, result = predict(img, model, 'cpu', class_name)
+        if display_prob:
+            print('Probability of {} : {:.6f}'.format(result, prob))
    
     return prob, result