edited error

app.py

@@ -106,7 +106,7 @@ def Final_Compute_regression_results_Sample(Model, batch_sampler,num_head=2):
             if ratios.vcdr < 0.6:
                 glaucoma = 'None (Rule of thumb ~0.6 or greater is suspicious)'
             else:
-                glaucoma = 'There May be a risk of Glaucoma (Rule of thumb ~0.6 or greater is suspicious)'
+                glaucoma = 'May be thre is a risk of Glaucoma (Rule of thumb ~0.6 or greater is suspicious)'
 
             # print('Galucoma:')
 
@@ -134,8 +134,8 @@ def infer(img):
     result, ratio, diagnosis, cropped = Final_Compute_regression_results_Sample(DeepLab, sample_batch, num_head=2)
 
     # cropped = cv2.cvtColor(np.asarray(cropped), cv2.COLOR_BGR2RGB)
-    cropped = result[cropped['cord'][0] -100 :cropped['cord'][1] +100,
-                     cropped['cord'][2] -100 :cropped['cord'][3] +100]
+    cropped = result[cropped['cord'][0] :cropped['cord'][1] ,
+                     cropped['cord'][2]  :cropped['cord'][3] ]
 
     return ratio, diagnosis, result, cropped
 
@@ -156,7 +156,7 @@ with gr.Blocks(css='#title {text-align : center;} ') as demo:
         )
     with gr.Row():
         with gr.Column():
-            prompt = gr.Image(label="Enter Your Retina Image")
+            prompt = gr.Image(label="Upload Your Retinal Fundus Image")
             btn = gr.Button(value='Submit')
             examples = gr.Examples(
                 ['M00027.png','M00056.png','M00073.png','M00093.png', 'M00018.png', 'M00034.png'],
@@ -165,8 +165,8 @@ with gr.Blocks(css='#title {text-align : center;} ') as demo:
             with gr.Row():
                 text1 = gr.Textbox(label="Vertical Cup to Disc Ratio:")
                 text2 = gr.Textbox(label="Predicted Diagnosis")
-            img = gr.Image(label='labeled image')
-            zoom = gr.Image(label='zoomed in')
+            img = gr.Image(label='Detected disc and cup')
+            zoom = gr.Image(label='Croppped')
 
             outputs = [text1,text2,img,zoom]
 

pipline.py

@@ -111,6 +111,7 @@ class ASPP(nn.Module):
         return x21
 
 
+
 class Transformer_Regression(nn.Module):
     def __init__(self, image_dim=224, dim_patch=24, num_classes=3, scale=1, feat_dim=192):
         super(Transformer_Regression, self).__init__()
@@ -164,8 +165,6 @@ class Transformer_Regression(nn.Module):
         return Score
 
 
-
-
 Ratios = namedtuple("Ratios", 'cdr hcdr vcdr')
 eps = np.finfo(np.float32).eps
 
@@ -219,4 +218,7 @@ def compute_ratios(mask_image):
     hcdr = (cup_horz + eps) / (disc_horz + eps)
     vcdr = (cup_vert + eps) / (disc_vert + eps)
 
-    return Ratios(cdr, hcdr, vcdr)
+    return Ratios(cdr, hcdr, vcdr)
+
+
+

requirements.txt

@@ -3,4 +3,6 @@ timm
 scikit-image
 numpy
 opencv-python
-Pillow
+Pillow
+gradio
+torchvision