Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ transform = transforms.Compose([
|
|
48 |
OBJECT_NAMES = ['enemies']
|
49 |
|
50 |
|
51 |
-
|
52 |
img_tensor = transform(image).unsqueeze(0)
|
53 |
orig_w, orig_h = image.size
|
54 |
|
|
|
48 |
OBJECT_NAMES = ['enemies']
|
49 |
|
50 |
|
51 |
+
def detect_objects_in_image(image):
|
52 |
img_tensor = transform(image).unsqueeze(0)
|
53 |
orig_w, orig_h = image.size
|
54 |
|