Spaces:

alkzar90
/

rock-glacier-segmentation

Runtime error

alkzar90 commited on Dec 18, 2022

Commit

da274a5

1 Parent(s): 9758099

fix typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,6 @@ def visualize_instance_seg_mask(mask):
 def query_image(img):
     """Función para generar predicciones a la escala origina"""
-    inputs = preprocessor(images=img, return_tensors="pt")
     inputs = preprocessor(img).unsqueeze(0)
     with torch.no_grad():
         preds = model(inputs)["logits"]

 def query_image(img):
     """Función para generar predicciones a la escala origina"""
     inputs = preprocessor(img).unsqueeze(0)
     with torch.no_grad():
         preds = model(inputs)["logits"]