Update app.py

app.py

@@ -8,7 +8,7 @@ class SkinCancer :
     def __init__ (self):
         self.model = self.load_model()
 
-    def build_model (self) :
+    def build_model () :
         model = Sequential()
         model.add(Conv2D(filters = 128, kernel_size = (4,4), input_shape = (32, 32, 3), activation = 'relu'))
         model.add(MaxPool2D(pool_size = (4,4)))
@@ -16,13 +16,11 @@ class SkinCancer :
         model.add(MaxPool2D(pool_size = (2,2)))
         model.add(BatchNormalization())
         #model.add(GlobalAveragePooling2D())
-
         model.add(Flatten())
         model.add(Dense(128, activation = 'relu'))
-        model.add(Dropout(0.1))
+        model.add(Dropout(0.2))
         model.add(Dense(2, activation = 'sigmoid')) # sigmoid is better for binary classification
-
-        #model.summary()
+        model.summary()
         return model
 
     def load_model(self):