Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -137,6 +137,7 @@ def transform_image(image):
|
|
| 137 |
|
| 138 |
# Definimos una funci贸n que se encarga de llevar a cabo las predicciones
|
| 139 |
def predict(img):
|
|
|
|
| 140 |
image = transforms.Resize((480,640))(img)
|
| 141 |
tensor = transform_image(image=image)
|
| 142 |
with torch.no_grad():
|
|
|
|
| 137 |
|
| 138 |
# Definimos una funci贸n que se encarga de llevar a cabo las predicciones
|
| 139 |
def predict(img):
|
| 140 |
+
img=Image.fromarray(img)
|
| 141 |
image = transforms.Resize((480,640))(img)
|
| 142 |
tensor = transform_image(image=image)
|
| 143 |
with torch.no_grad():
|