v-1

app.py

@@ -83,10 +83,10 @@ def UNet2_with_STN():
     model = Model(inputs, outputs)
     return model
 
-    model = UNet2_with_STN()
-    mean_iou = MeanIoU(num_classes=2)
-    model.compile(optimizer="adam", loss="binary_crossentropy", metrics=["acc",mean_iou])
-    model.load_weights('/content/drive/MyDrive/B3_cell_seg/UNet+stn_model_longrun.h5')
+model = UNet2_with_STN()
+mean_iou = MeanIoU(num_classes=2)
+model.compile(optimizer="adam", loss="binary_crossentropy", metrics=["acc",mean_iou])
+model.load_weights('./model_file/UNet+stn_model_longrun.h5')
 
 def cellsegmentor(img):