Update app.py

app.py

@@ -128,8 +128,8 @@ def predict(radio, dict, word_mask, prompt=""):
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
         filename = f"{uuid.uuid4()}.png"
         plt.imsave(filename,torch.sigmoid(preds[0][0]))
-        ret,img2 = cv2.imread(filename)
-        if ret == True: 
+        img2 = cv2.imread(filename)
+        #if ret == True: 
             gray_image = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
             (thresh, bw_image) = cv2.threshold(gray_image, 100, 255, cv2.THRESH_BINARY)
             cv2.cvtColor(bw_image, cv2.COLOR_BGR2RGB)
@@ -158,9 +158,9 @@ def predict(radio, dict, word_mask, prompt=""):
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
         filename = f"{uuid.uuid4()}.png"
         plt.imsave(filename,torch.sigmoid(preds[0][0]))
-        ret,img2 = cv2.imread(filename)
+        img2 = cv2.imread(filename)
         
-        if ret == True: 
+        #if ret == True: 
             gray_image = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
             (thresh, bw_image) = cv2.threshold(gray_image, 100, 255, cv2.THRESH_BINARY)
             cv2.cvtColor(bw_image, cv2.COLOR_BGR2RGB)