Spaces:

aioverlords-amnil
/

OCR-SMALL

Build error

bikram commited on May 12

Commit

d08383c

1 Parent(s): 57db2f7

label encoded to string

Files changed (2) hide show

__pycache__/utils.cpython-310.pyc CHANGED Viewed

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

utils.py CHANGED Viewed

@@ -389,4 +389,9 @@ def predict_ne(image_path, device="cpu"):
             output = model(image_tensor)
             _, predicted = torch.max(output, 1)
         print(predicted.item())
-        return predicted.item()

             output = model(image_tensor)
             _, predicted = torch.max(output, 1)
         print(predicted.item())
+        if predicted.item() == 0:
+            return "Nepali"
+        elif predicted.item() == 1:
+            return "English"
+        else:
+         return predicted.item()