ysharma HF Staff commited on
Commit
473da3b
·
1 Parent(s): 25e59a3
Files changed (1) hide show
  1. app.py +54 -0
app.py ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from PIL import Image
3
+
4
+ demo_inf = gr.Blocks.load(name="spaces/ICML2022/OFA") #"spaces/freddyaboulton/blocks_inputs")
5
+ #assert demo_inf("Foo", "bar") == "Foo bar"
6
+
7
+ def txt_fun(txt1, txt2):
8
+ return demo_inf(txt1, txt2)
9
+
10
+ #original = Image.open("./c4.jpg")
11
+ #output_img_path = demo_inf("./c4.jpg", fn_index=1)
12
+ #output = Image.open(output_img_path)
13
+ #assert original.size == output.size
14
+
15
+ def img_fun1(img_in):
16
+ return demo_inf(img_in) #, fn_index=1)
17
+
18
+ #original = Image.open("./c4.jpg")
19
+ #output_img_path = demo_inf("./c4.jpg", fn_index=1)
20
+ #output = Image.open(output_img_path)
21
+ #assert original.size == output.size
22
+
23
+ def img_fun(img_in):
24
+ original = Image.open("./c4.jpg")
25
+ print("after line1")
26
+ output_img_path = demo_inf("./c4.jpg", fn_index=1)
27
+ print("after line1")
28
+ output = Image.open(output_img_path)
29
+ print("after line1")
30
+ return output
31
+
32
+ demo = gr.Blocks()
33
+
34
+ with demo:
35
+ gr.Markdown("<h1><center>Testing</center></h1>")
36
+ gr.Markdown(
37
+ """Testing Inference for Gradio. Work in Progress."""
38
+ )
39
+ with gr.Row():
40
+ in_image = gr.Image(visible=False)
41
+ out_image = gr.Image()
42
+
43
+ in_text1 = gr.Textbox()
44
+ in_text2 = gr.Textbox()
45
+ out_text = gr.Textbox()
46
+
47
+ b1 = gr.Button("Image Button")
48
+ b2 = gr.Button("Text Button")
49
+
50
+ b1.click(img_fun,inputs=in_image, outputs=out_image)
51
+ b2.click(txt_fun, inputs=[in_text1, in_text2], outputs=out_text)
52
+ #examples=examples
53
+
54
+ demo.launch(enable_queue=True, debug=True)