Change interface

app.py

@@ -53,7 +53,6 @@ def init_huggingface():
     )
     return pretrained_path
 
-@torch.no_grad()
 @spaces.GPU
 def init_distributed(opt):
     opt['CUDA'] = opt.get('CUDA', True) and torch.cuda.is_available()
@@ -88,10 +87,10 @@ def init_distributed(opt):
 def setup_model():
     """Initialize the model on CPU without CUDA initialization."""
     opt = load_opt_from_config_files(["configs/biomedparse_inference.yaml"])
-    opt['CUDA'] = opt.get('CUDA', True) and torch.cuda.is_available()  # Vérifier la disponibilité de CUDA
     opt = init_distributed(opt)
+    opt['device'] = 'cuda' if torch.cuda.is_available() else 'cpu'
     pretrained_path = init_huggingface()
-    model = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_path)
+    model = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_path).eval()
     return model
 
 @torch.no_grad()
@@ -146,6 +145,7 @@ def predict_image(model, image, prompts):
                                    mode='bilinear')[0,:,:data['height'],:data['width']].sigmoid().cpu().detach().numpy()
     return pred_mask_prob
 
+
 def process_image(image, prompts, model):
     """Process image with proper error handling."""
     try:

modeling/architectures/seem_model_demo.py

@@ -134,8 +134,9 @@ class GeneralizedSEEM(nn.Module):
         interactive_mode = 'best'
         interactive_iter = 20
         dilation = 3
-        dilation_kernel = torch.ones((1, 1, dilation, dilation), device=torch.cuda.current_device())
-
+        device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        dilation_kernel = torch.ones((1, 1, dilation, dilation), device=device)
+        
         return {
             "backbone": backbone,
             "sem_seg_head": sem_seg_head,

requirements.txt

@@ -1,5 +1,4 @@
 pillow==9.4.0
-accelerate
 opencv-python==4.8.1.78
 pyyaml==6.0.1
 json_tricks==3.17.3
@@ -15,7 +14,6 @@ sentencepiece==0.1.99
 ftfy==6.1.1
 regex==2023.10.3
 nltk==3.8.1
-#mpi4py
 vision-datasets==0.2.2
 cython==3.0.2
 pycocotools==2.0.7