Update app.py

app.py

@@ -1,4 +1,6 @@
+import gradio as gr
 import os 
+import sys
 os.system("mkdir pose2pose")
 os.system("cd pose2pose")
 os.system("mkdir -p datasets/DeepFashion")
@@ -6,16 +8,24 @@ os.system("mkdir -p output/DeepFashion/ckpt/pretrained")
 os.system("git clone https://github.com/prasunroy/pose-transfer.git")
 os.system("cd pose-transfer")
 os.system("pip install -r requirements.txt")
-import sys
 sys.path.append("pose-transfer")
 import torch 
-
 from api import Pose2Pose
 from PIL import Image
 
 p2p = Pose2Pose(pretrained=True)
+def ifer(con_im,ref_im):
+    condition = Image.open(con_im)
+    reference = Image.open(ref_im)
+    generated = p2p.transfer_as(condition, reference)
+    #generated.show()
+    return generated
+with gr.Blocks() as app:
+    with gr.Row():
+        condition_im = gr.Image(type='filepath')
+        reference_im = gr.Image(type='filepath')
+    btn=gr.Button()
+    output_im = gr.Image()
+    btn.click(infer,[condition_im,reference_im], output_im)
 
-condition = Image.open('./api-test/condition.jpg')
-reference = Image.open('./api-test/target_pose_reference.jpg')
-generated = p2p.transfer_as(condition, reference)
-generated.show()
+app.launch()