Upload handler.py with huggingface_hub

handler.py

@@ -1,3 +1,16 @@
+import subprocess
+import sys
+import os
+
+# Install transformers from source BEFORE any transformers import
+if not os.path.exists("/tmp/.transformers_installed"):
+    subprocess.check_call([
+        sys.executable, "-m", "pip", "install", "-q", "--upgrade",
+        "git+https://github.com/huggingface/transformers.git",
+        "accelerate"
+    ])
+    open("/tmp/.transformers_installed", "w").close()
+
 import torch
 import base64
 import io
@@ -28,7 +41,7 @@ class EndpointHandler:
             image_b64 = inputs_data
             prompt = "Text Recognition:"
         else:
-            return [{"error": "Send {inputs: {image: base64_string, prompt: str}}"}]
+            return [{"error": "Send {inputs: {image: base64, prompt: str}}"}]
 
         try:
             image_bytes = base64.b64decode(image_b64)