Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ model.eval()
|
|
37 |
# transforms.ToTensor()
|
38 |
# ])
|
39 |
transform = transforms.Compose([ # Ensure input is a PIL image
|
40 |
-
transforms.Resize((
|
41 |
transforms.ToTensor()
|
42 |
])
|
43 |
# transform = transforms.Compose([
|
@@ -85,6 +85,7 @@ def detect_objects_in_image(image):
|
|
85 |
print(Image.fromarray(np.array(image)))
|
86 |
print(np.array(image))
|
87 |
print(type(image))
|
|
|
88 |
if len(pred) == 0:
|
89 |
return Image.fromarray(np.array(image)) # Return only image and None for graph
|
90 |
print("passed6")
|
|
|
37 |
# transforms.ToTensor()
|
38 |
# ])
|
39 |
transform = transforms.Compose([ # Ensure input is a PIL image
|
40 |
+
transforms.Resize((1000, 1000)),
|
41 |
transforms.ToTensor()
|
42 |
])
|
43 |
# transform = transforms.Compose([
|
|
|
85 |
print(Image.fromarray(np.array(image)))
|
86 |
print(np.array(image))
|
87 |
print(type(image))
|
88 |
+
print(len(pred))
|
89 |
if len(pred) == 0:
|
90 |
return Image.fromarray(np.array(image)) # Return only image and None for graph
|
91 |
print("passed6")
|