Update app.py

app.py

@@ -462,6 +462,25 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             final_image = img  # Update final_image with the current image
         return final_image
 
+def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
+    pipe_i2i.to("cuda")
+    generator = torch.Generator(device="cuda").manual_seed(seed)
+    image_input = load_image(image_input_path)
+    final_image = pipe_i2i(
+        prompt=prompt_mash,
+        image=image_input,
+        strength=image_strength,
+        num_inference_steps=steps,
+        guidance_scale=cfg_scale,
+        width=width,
+        height=height,
+        generator=generator,
+        joint_attention_kwargs={"scale": 1.0},
+        output_type="pil",
+    ).images[0]
+    return final_image
+
+@spaces.GPU(duration=75)
 def run_lora(prompt, cfg_scale, steps, selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, randomize_seed, seed, width, height, loras_state, image_input=None, progress=gr.Progress(track_tqdm=True)):
     print("run_lora function called.")  # Debugging statement
     print(f"Inputs received - Prompt: {prompt}, CFG Scale: {cfg_scale}, Steps: {steps}, Seed: {seed}, Width: {width}, Height: {height}")  # Debugging statement