Update app.py

app.py

@@ -59,25 +59,10 @@ def predict_image(image, model, device, transform, threshold=0.5):
     label_names = [map_label_to_class(label.item()) for label in labels]
     drawn_image = draw_boxes_with_labels(image_uint8, boxes.cpu(), label_names)
 
-    return drawn_image.permute(1, 2, 0)  # Đảm bảo định dạng hình ảnh là (H, W, C)
-
-# Định nghĩa hàm get_transforms
-def get_transforms(train=False):
-    if train:
-        transform = A.Compose([
-            A.Resize(800, 800),  # Kích thước đầu vào 800px
-            A.HorizontalFlip(p=0.3),
-            A.VerticalFlip(p=0.3),
-            A.RandomBrightnessContrast(p=0.1),
-            A.ColorJitter(p=0.1),
-            ToTensorV2()
-        ], bbox_params=A.BboxParams(format='coco'))
-    else:
-        transform = A.Compose([
-            A.Resize(800, 800),  # Kích thước đầu vào 800px
-            ToTensorV2()
-        ], bbox_params=A.BboxParams(format='coco'))
-    return transform
+    # Convert tensor to PIL image for Gradio compatibility
+    drawn_image_pil = F.to_pil_image(drawn_image)
+    
+    return np.array(drawn_image_pil)  # Convert PIL image to NumPy array
 
 # Tải mô hình
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -110,4 +95,4 @@ iface = gr.Interface(
 
 # Chạy ứng dụng
 if __name__ == "__main__":
-    iface.launch(share=True)  # Set share=True to create a public link
+    iface.launch()