bram-w commited on
Commit
cb1cb75
1 Parent(s): 1a4ea1a

copy local version so can modify app.py with requests

Browse files
Files changed (1) hide show
  1. local_app.py +33 -0
local_app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import numpy as np
3
+ from edict_functions import EDICT_editing
4
+ from PIL import Image
5
+
6
+ def greet(name):
7
+ return "Hello " + name + "!!"
8
+
9
+
10
+ def edict(x, source_text, edit_text,
11
+ edit_strength, guidance_scale,
12
+ steps=50, mix_weight=0.93, ):
13
+ x = Image.fromarray(x)
14
+ return_im = EDICT_editing(x,
15
+ source_text,
16
+ edit_text,
17
+ steps=steps,
18
+ mix_weight=mix_weight,
19
+ init_image_strength=edit_strength,
20
+ guidance_scale=guidance_scale
21
+ )[0]
22
+ return np.array(return_im)
23
+
24
+ iface = gr.Interface(fn=edict, inputs=["image",
25
+ gr.Textbox(label="Original Description"),
26
+ gr.Textbox(label="Edit Description"),
27
+ # 50, # gr.Slider(5, 50, value=20, step=1),
28
+ # 0.93, # gr.Slider(0.5, 1, value=0.7, step=0.05),
29
+ gr.Slider(0.0, 1, value=0.8, step=0.05),
30
+ gr.Slider(0, 10, value=3, step=0.5),
31
+ ],
32
+ outputs="image")
33
+ iface.launch()