Spaces:

ConradLax
/

document_classifier

Runtime error

ConradLax commited on Dec 18, 2023

Commit

8c40583

1 Parent(s): 41d3948

fix: error in opening url image

Files changed (1) hide show

main.py CHANGED Viewed

@@ -135,8 +135,9 @@ import os
 import requests
 def classify_acct_dtype_str(input_path):
-    response = requests.get("https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/city-streets.jpg")
-    dtype_inf, dtype_conf = doctype_classify(response, "city-streets.jpg")
     return dtype_inf

 import requests
 def classify_acct_dtype_str(input_path):
+    # response = requests.get("https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/city-streets.jpg")
+    ipt = Image.open(input_path)
+    dtype_inf, dtype_conf = doctype_classify(ipt, "city-streets.jpg")
     return dtype_inf