Update app.py

app.py

@@ -38,20 +38,20 @@ class GeoTr_Seg(nn.Module):
 
 # Initialize models
 GeoTr_Seg_model = GeoTr_Seg()
-IllTr_model = IllTr()
+#IllTr_model = IllTr()
 
 # Load models only once
 reload_segmodel(GeoTr_Seg_model.msk, './model_pretrained/seg.pth')
 reload_model(GeoTr_Seg_model.GeoTr, './model_pretrained/geotr.pth')
-reload_model(IllTr_model, './model_pretrained/illtr.pth')
+#reload_model(IllTr_model, './model_pretrained/illtr.pth')
 
 # Compile models (assuming PyTorch 2.0)
 GeoTr_Seg_model = torch.compile(GeoTr_Seg_model)
-IllTr_model = torch.compile(IllTr_model)
+#IllTr_model = torch.compile(IllTr_model)
 
 def process_image(input_image):
     GeoTr_Seg_model.eval()
-    IllTr_model.eval()
+    #IllTr_model.eval()
 
     im_ori = np.array(input_image)[:, :, :3] / 255.
     h, w, _ = im_ori.shape