1

app.py

@@ -1,11 +1,17 @@
 import gradio as gr
 from PIL import Image
 
-demo_inf = gr.Blocks.load(name="spaces/ICML2022/OFA") #"spaces/freddyaboulton/blocks_inputs")
+demo_inf_ofa = gr.Interface.load(name="spaces/ICML2022/OFA") #"spaces/freddyaboulton/blocks_inputs")
+demo_inf_lama = gr.Interface.load(name="spaces/CVPR/lama-example")  #gr.Blocks.load
+#img = gr.Interface.load("spaces/multimodalart/latentdiffusion")(ent[0],'50','256','256','1',10)[0] 
+
 #assert demo_inf("Foo", "bar") == "Foo bar"
 
-def txt_fun(txt1, txt2):
-  return demo_inf(txt1, txt2)
+def get_lama(img1, img2):
+  out = demo_inf_lama()(img1, img2, "automatic (U2net)")
+  
+#def txt_fun(txt1, txt2):
+#  return demo_inf(txt1, txt2)
 
 #original = Image.open("./c4.jpg")
 #output_img_path = demo_inf("./c4.jpg", fn_index=1)
@@ -37,18 +43,19 @@ with demo:
         """Testing Inference for Gradio. Work in Progress."""
         )
   with gr.Row():
-    in_image = gr.Image(visible=False)
+    in_image = gr.Image() #(visible=False)
+    in_image_mask = gr.Image(visible=False)
     out_image = gr.Image()
     
-    in_text1 = gr.Textbox()
-    in_text2 = gr.Textbox()
-    out_text = gr.Textbox()
+    #in_text1 = gr.Textbox()
+    #in_text2 = gr.Textbox()
+    #out_text = gr.Textbox()
     
   b1 = gr.Button("Image Button")
-  b2 = gr.Button("Text Button") 
+  #b2 = gr.Button("Text Button") 
 
-  b1.click(img_fun,inputs=in_image, outputs=out_image) 
-  b2.click(txt_fun, inputs=[in_text1, in_text2], outputs=out_text)
+  b1.click(get_lama,inputs=[in_image, in_image_mask], outputs=out_image) 
+  #b2.click(txt_fun, inputs=[in_text1, in_text2], outputs=out_text)
   #examples=examples 
 
-demo.launch(enable_queue=True, debug=True)
+demo.launch(enable_queue=True, debug=True)