Update app.py

app.py

@@ -416,15 +416,18 @@ def my_inference_function(image,text):
   img_dir = tempfile.mkdtemp()
     
   img_path = os.path.join(img_dir, f"out_112.jpeg")
-  decodedData = base64.b64decode((image))
+  #decodedData = base64.b64decode((image))
 
   print("----- my_inference_function img_path -----")
   print(img_path)
     
   # Write Image from Base64 File
-  imgFile = open(img_path, 'wb')
-  imgFile.write(decodedData)
-  imgFile.close()
+  #imgFile = open(img_path, 'wb')
+  #imgFile.write(decodedData)
+  #imgFile.close()
+
+  with open(img_path, "wb") as fh:
+    fh.write(base64.urlsafe_b64decode(image))
     
   return editor.predict(img_path,[text])