commit

app.py

@@ -70,21 +70,19 @@ def process_image_4(image_path, prompt):
 
         w, h = image.size
 
-        coor_point = torch.zeros((1,5,2)).to(torch.float32)
-        point_labels = torch.zeros((1,5,1)).to(torch.float32)
+        coor_point = []
+        point_labels = []
 
         image = image.resize((768, 768), Image.LANCZOS).convert('RGB')
-        to_tensor = transforms.ToTensor()
-        image = (to_tensor(image) - 0.5) * 2
+        
 
         cur_input = {
-                'input_images': image.unsqueeze(0),
-                'original_size': torch.tensor([[w,h]]),
-                'target_size': torch.tensor([[768, 768]]),
+                'input_images': image,
+                'original_size': [[w,h]],
+                'target_size': [[768, 768]],
                 'prompt': [cur_p],
                 'coor_point': coor_point,
                 'point_labels': point_labels,
-                'generator': generator
             }
         inputs.append(cur_input)