fffiloni commited on
Commit
f6dbf53
1 Parent(s): 7a3b159

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -57,18 +57,18 @@ def infer(image_in, camera_shot, conditional_pose, prompt, chosen_model):
57
  else :
58
  raise gr.Error("No custom conditional shot found !")
59
 
60
- elif camera_shot == "close-up" :
61
- #
62
  elif camera_shot == "medium close-up":
63
- #
64
  elif camera_shot == "medium shot":
65
- #
66
  elif camera_shot == "cowboy shot":
67
- #
68
  elif camera_shot == "medium full shot":
69
- #
70
  elif camera_shot == "full shot":
71
- #
72
 
73
  iid_img = get_instantID(image_in, conditional_pose, prompt)
74
 
 
57
  else :
58
  raise gr.Error("No custom conditional shot found !")
59
 
60
+ elif camera_shot == "close-up":
61
+ conditional_pose = "camera_shots/close_up_shot.jpeg"
62
  elif camera_shot == "medium close-up":
63
+ conditional_pose = "camera_shots/medium_close_up.jpeg"
64
  elif camera_shot == "medium shot":
65
+ conditional_pose = "camera_shots/medium_shot.png"
66
  elif camera_shot == "cowboy shot":
67
+ conditional_pose = "camera_shots/cowboy_shot.jpeg"
68
  elif camera_shot == "medium full shot":
69
+ conditional_pose = "camera_shots/medium_full_shot.png"
70
  elif camera_shot == "full shot":
71
+ conditional_pose = "camera_shots/full_shot.jpeg"
72
 
73
  iid_img = get_instantID(image_in, conditional_pose, prompt)
74