Update app.py

app.py

@@ -6,9 +6,9 @@ hf_token = os.environ.get("HF_TKN")
 
 from style_template import styles
 STYLE_NAMES = list(styles.keys())
-DEFAULT_STYLE_NAME = "Watercolor"
+DEFAULT_STYLE_NAME = "(No Style)"
 
-def get_instantID(portrait_in, condition_pose, prompt):
+def get_instantID(portrait_in, condition_pose, prompt, style):
     client = Client("https://fffiloni-instantid.hf.space/", hf_token=hf_token)
     negative_prompt = "(lowres, low quality, worst quality:1.2), (text:1.2), watermark, (frame:1.2), deformed, ugly, deformed eyes, blur, out of focus, blurry, deformed cat, deformed, photo, anthropomorphic cat, monochrome, photo, pet collar, gun, weapon, blue, 3d, drones, drone, buildings in background, green"
     result = client.predict(
@@ -16,7 +16,7 @@ def get_instantID(portrait_in, condition_pose, prompt):
 		condition_pose,	# filepath  in 'Upload a reference pose image (optional)' Image component
 	    prompt,	# str  in 'Prompt' Textbox component
 		negative_prompt,	# str  in 'Negative Prompt' Textbox component
-		"(No style)",	# Literal['(No style)', 'Watercolor', 'Film Noir', 'Neon', 'Jungle', 'Mars', 'Vibrant Color', 'Snow', 'Line art']  in 'Style template' Dropdown component
+		style,	# Literal['(No style)', 'Watercolor', 'Film Noir', 'Neon', 'Jungle', 'Mars', 'Vibrant Color', 'Snow', 'Line art']  in 'Style template' Dropdown component
 		True,	# bool  in 'Enhance non-face region' Checkbox component
 		20,	# float (numeric value between 20 and 100) in 'Number of sample steps' Slider component
 		0.8,	# float (numeric value between 0 and 1.5) in 'IdentityNet strength (for fedility)' Slider component
@@ -54,7 +54,7 @@ def get_video_svd(image_in):
     print(result)
     return result[0]["video"]
 
-def infer(image_in, camera_shot, conditional_pose, prompt, chosen_model):
+def infer(image_in, camera_shot, conditional_pose, prompt, style, chosen_model):
     if camera_shot == "custom":
         if conditional_pose != None:
             conditional_pose = conditional_pose
@@ -74,7 +74,7 @@ def infer(image_in, camera_shot, conditional_pose, prompt, chosen_model):
     elif camera_shot == "full shot":
         conditional_pose = "camera_shots/full_shot.jpeg"
     
-    iid_img = get_instantID(image_in, conditional_pose, prompt)
+    iid_img = get_instantID(image_in, conditional_pose, prompt, style)
     
     if chosen_model == "i2vgen-xl" :
         video_res = get_video_i2vgen(iid_img, prompt)
@@ -134,6 +134,7 @@ with gr.Blocks(css=css) as demo:
             camera_shot,
             condition_shot,
             prompt,
+            style,
             chosen_model
         ],
         outputs = [