Update app.py

app.py

@@ -236,7 +236,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             
     if image_input is not None:
         final_image, file_path, base64_str = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, lora_scale, seed)
-        yield final_image, seed, file_path, base64_str, gr.update(value=file_path, visible=True)
+        return final_image, file_path, base64_str
     else:
         image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress)
     
@@ -255,7 +255,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         with open(temp_file.name, "rb") as f:
             img_base64 = base64.b64encode(f.read()).decode("utf-8")
 
-        yield final_image, seed, temp_file.name, f"data:image/png;base64,{img_base64}", gr.update(value=temp_file.name, visible=True)
+        return final_image, temp_file.name, base64_str
         
 css = '''
 #gen_btn{height: 100%}