Omnibus commited on
Commit
c48e116
·
1 Parent(s): 4e50bb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -5,4 +5,15 @@ os.system("mkdir -p datasets/DeepFashion")
5
  os.system("mkdir -p output/DeepFashion/ckpt/pretrained")
6
  os.system("git clone https://github.com/prasunroy/pose-transfer.git")
7
  os.system("cd pose-transfer")
8
- os.system("pip install -r requirements.txt")
 
 
 
 
 
 
 
 
 
 
 
 
5
  os.system("mkdir -p output/DeepFashion/ckpt/pretrained")
6
  os.system("git clone https://github.com/prasunroy/pose-transfer.git")
7
  os.system("cd pose-transfer")
8
+ os.system("pip install -r requirements.txt")
9
+
10
+
11
+ from api import Pose2Pose
12
+ from PIL import Image
13
+
14
+ p2p = Pose2Pose(pretrained=True)
15
+
16
+ condition = Image.open('./api-test/condition.jpg')
17
+ reference = Image.open('./api-test/target_pose_reference.jpg')
18
+ generated = p2p.transfer_as(condition, reference)
19
+ generated.show()