fix typo
Browse files
app.py
CHANGED
|
@@ -45,7 +45,6 @@ def visualize_instance_seg_mask(mask):
|
|
| 45 |
|
| 46 |
def query_image(img):
|
| 47 |
"""Función para generar predicciones a la escala origina"""
|
| 48 |
-
inputs = preprocessor(images=img, return_tensors="pt")
|
| 49 |
inputs = preprocessor(img).unsqueeze(0)
|
| 50 |
with torch.no_grad():
|
| 51 |
preds = model(inputs)["logits"]
|
|
|
|
| 45 |
|
| 46 |
def query_image(img):
|
| 47 |
"""Función para generar predicciones a la escala origina"""
|
|
|
|
| 48 |
inputs = preprocessor(img).unsqueeze(0)
|
| 49 |
with torch.no_grad():
|
| 50 |
preds = model(inputs)["logits"]
|