linoyts HF Staff commited on
Commit
78f9ab5
·
verified ·
1 Parent(s): e0e61bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -174,6 +174,8 @@ def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, r
174
  trigger_word = lora_to_use["trigger_word"]
175
  if trigger_word == ", How2Draw":
176
  prompt = f"create a How2Draw sketch of the person of the photo {prompt}, maintain the facial identity of the person and general features"
 
 
177
  else:
178
  prompt = f"convert the style of this portrait photo to {trigger_word} while maintaining the identity of the person. {prompt}. Make sure to maintain the person's facial identity and features, while still changing the overall style to {trigger_word}."
179
 
 
174
  trigger_word = lora_to_use["trigger_word"]
175
  if trigger_word == ", How2Draw":
176
  prompt = f"create a How2Draw sketch of the person of the photo {prompt}, maintain the facial identity of the person and general features"
177
+ elif trigger_word == ", video game screenshot in the style of THSMS":
178
+ prompt = f"create a video game screenshot in the style of THSMS with the person from the photo, {prompt}. maintain the facial identity of the person and general features"
179
  else:
180
  prompt = f"convert the style of this portrait photo to {trigger_word} while maintaining the identity of the person. {prompt}. Make sure to maintain the person's facial identity and features, while still changing the overall style to {trigger_word}."
181