Update app_v4.py

app_v4.py

@@ -42,7 +42,7 @@ except Exception as e:
     print(f"Error setting memory usage: {e}")
 
 text_encoder_2_unquant = T5EncoderModel.from_pretrained(
-    "LPX55/FLUX.1-merged_uncensored",
+    "LPX55/FLUX.1-merged_lightning_v2",
     subfolder="text_encoder_2",
     torch_dtype=torch.bfloat16,
     token=huggingface_token
@@ -269,7 +269,7 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as demo:
         ],
         outputs=[log_state, generated_image, prompt]
     )
-    control_image.input(
+    control_image.upload(
         generate_caption,
         inputs=[control_image],
         outputs=[caption_state]