Update app.py

app.py

@@ -413,29 +413,36 @@ with blocks:
 
 
 def my_inference_function(image,text):
-  img_dir = tempfile.mkdtemp()
+  #img_dir = tempfile.mkdtemp()
     
   #img_path = os.path.join(img_dir, f"_112.jpeg")
-  img_path ="/tmp/111_112.jpeg"
+  #img_path ="/tmp/111_112.jpeg"
   #decodedData = base64.b64decode((image))
 
-  print("----- my_inference_function img_path -----")
-  print(img_path)
+  #print("----- my_inference_function img_path -----")
+  #print(img_path)
     
   # Write Image from Base64 File
   #imgFile = open(img_path, 'wb')
   #imgFile.write(decodedData)
   #imgFile.close()
 
-  with open(img_path, "wb") as fh:
-    fh.write(base64.urlsafe_b64decode(image))
+  #with open(img_path, "wb") as fh:
+  #  fh.write(base64.urlsafe_b64decode(image))
 
-  file = open(img_path, 'rb')
-  encoded_data = file.read()
-  file.close()
+  #file = open(img_path, 'rb')
+  #encoded_data = file.read()
+  #file.close()
 
-  print("----- my_inference_function encoded_data -----")
-  print(encoded_data)
+  temp = tempfile.NamedTemporaryFile()
+  temp.write(image)
+  print("Name of the file is:", temp.name)
+  temp.close()
+
+  img_path = temp.name
+
+  print("----- my_inference_function temp.read -----")
+  print(temp.read())
       
   return editor.predict(img_path,[text],edit_choices={'edit_type': 'None', 'pose': 0, 'smile': 0, 'gender': 0, 'age': 0, 'hair_length': 0, 'src_text': '', 'tar_text': '', 'alpha': 1, 'beta': 0.14})