test gradio

app.py

@@ -115,7 +115,7 @@ This way, each frame represents a distinct scene, and there’s no redundancy be
     return conditioning_frames
 
 @spaces.GPU
-def generate_gif(prompt, reference_image, controlnet_conditioning_scale,num_frames):
+def generate_gif(prompt, reference_image, controlnet_conditioning_scale,style_conditioning_scale,num_frames):
     massage_history = [{"role": "system", "content": """
 You are a scene designer tasked with creating sparse frames of a video. You will be given a prompt describing the desired video, and your goal is to design only the key frames (sparse frames) that represent major changes in the scene. Do not include repetitive or similar scenes—only capture distinct moments.
 
@@ -144,7 +144,7 @@ This way, each frame represents a distinct scene, and there’s no redundancy be
     conditioning_frames = []
     controlnet_frame_indices =[]
     for frame in frames:
-        conditioning_frames.append(generate_image(frame['description'], reference_image, float(controlnet_conditioning_scale)))
+        conditioning_frames.append(generate_image(frame['description'], reference_image, float(style_conditioning_scale)))
         controlnet_frame_indices.append(frame['frame_index'])
         yield conditioning_frames, "animation.gif"
 
@@ -155,6 +155,7 @@ This way, each frame represents a distinct scene, and there’s no redundancy be
         conditioning_frames=conditioning_frames,
         controlnet_frame_indices=controlnet_frame_indices,
         controlnet_conditioning_scale=float(controlnet_conditioning_scale),
+        num_frames=max(controlnet_frame_indices+1),
     ).frames[0]
     export_to_gif(video, "output.gif")
 
@@ -168,6 +169,7 @@ interface = gr.Interface(
         # gr.Image( type= "filepath",label="Reference Image (Style)"),
         gr.File(type="filepath",file_count="multiple",label="Reference Image (Style)"),
         gr.Slider(label="Control Net Conditioning Scale", minimum=0, maximum=1.0, step=0.1, value=1.0),
+        gr.Slider(label="Style Scale", minimum=0, maximum=1.0, step=0.1, value=1.0),
         gr.Slider(label="Number of frames", minimum=0, maximum=100.0, step=1.0, value=10.0),
 
     ],