Update app.py

app.py

@@ -2,12 +2,11 @@ import gradio as gr
 import numpy as np
 from edict_functions import EDICT_editing
 from PIL import Image
+from utils import Endpoint, get_token
 
-def greet(name):
-    return "Hello " + name + "!!"
 
 
-def edict(x, source_text, edit_text,
+def local_edict(x, source_text, edit_text,
          edit_strength, guidance_scale,
           steps=50, mix_weight=0.93, ):
     x = Image.fromarray(x)
@@ -21,6 +20,42 @@ def edict(x, source_text, edit_text,
                               )[0]
     return np.array(return_im)
 
+
+
+def decode_image(img_obj):
+    img = Image.open(img_obj).convert("RGB")
+    return img
+
+def edict(x, source_text, edit_text,
+         edit_strength, guidance_scale,
+          steps=50, mix_weight=0.93, ):
+
+    url = endpoint.url
+    url = url + "/api/edit"
+    headers = {
+        "User-Agent": "EDICT HuggingFace Space",
+        "Auth-Token": get_token(),
+    }
+
+    data = {
+        "source_text": source_text,
+        "edit_text": edit_text,
+        "edit_strength": edit_strength,
+        "guidance_scale": guidance_scale,
+    }
+
+    image = encode_image(x)
+    files = {"image": image}  
+
+    response = requests.post(url, data=data, files=files, headers=headers)
+    
+    if response.status_code == 200:
+        return np.array(decode_image(BytesIO(response.content)))
+    else:
+        return "Error: " + response.text   
+    # x = decode_image(response)
+    # return np.array(x)
+
 iface = gr.Interface(fn=edict, inputs=["image",
                                        gr.Textbox(label="Original Description"),
                                        gr.Textbox(label="Edit Description"),
@@ -30,4 +65,4 @@ iface = gr.Interface(fn=edict, inputs=["image",
                                        gr.Slider(0, 10, value=3, step=0.5),
                                       ],
                      outputs="image")
-iface.launch()
+iface.launch()