Update app.py

app.py

@@ -23,6 +23,8 @@ from styleclip.styleclip_global import project_code_with_styleclip, style_tensor
 
 import clip
 
+import base64
+
 model_dir = "models"
 os.makedirs(model_dir, exist_ok=True)
 
@@ -411,7 +413,20 @@ with blocks:
 
 
 def my_inference_function(image,text):
-  return editor.predict(image,[text])
+  img_dir = tempfile.mkdtemp()
+    
+  img_path = os.path.join(img_dir, f"out_111.jpg")
+  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()
+    
+  return editor.predict(img_path,[text])
 
 gradio_interface = gr.Interface(
   fn = my_inference_function,