Spaces:

AndrewKof
/

NEMOtools

Running

AndrewKof commited on Nov 19, 2025

Commit

a47fe92

1 Parent(s): 3d09155

🔥 Fix NumPy compatibility + processor padding

Files changed (4) hide show

app/__pycache__/inference.cpython-310.pyc CHANGED Viewed

Binary files a/app/__pycache__/inference.cpython-310.pyc and b/app/__pycache__/inference.cpython-310.pyc differ

app/__pycache__/main.cpython-310.pyc CHANGED Viewed

Binary files a/app/__pycache__/main.cpython-310.pyc and b/app/__pycache__/main.cpython-310.pyc differ

app/__pycache__/model.cpython-310.pyc CHANGED Viewed

Binary files a/app/__pycache__/model.cpython-310.pyc and b/app/__pycache__/model.cpython-310.pyc differ

app/inference.py CHANGED Viewed

@@ -161,7 +161,7 @@ def classify_bytes(image_bytes: bytes):
     # Load and preprocess image
     image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
-    inputs = processor(images=image, return_tensors="pt").to(DEVICE)
     # Forward pass
     with torch.no_grad():

     # Load and preprocess image
     image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+    inputs = processor(images=image, return_tensors="pt", padding="True").to(DEVICE)
     # Forward pass
     with torch.no_grad():