diff --git a/Mflux_Comfy/Mflux_Air.py b/Mflux_Comfy/Mflux_Air.py
index ae1167c..565cd0a 100644
--- a/Mflux_Comfy/Mflux_Air.py
+++ b/Mflux_Comfy/Mflux_Air.py
@@ -4,7 +4,7 @@ from huggingface_hub import snapshot_download
 from folder_paths import models_dir 
 from mflux.config.model_config import ModelConfig
 from mflux.flux.flux import Flux1
-from .Mflux_Core import generate_image
+from .Mflux_Core import get_lora_info, generate_image
 
 def create_directory(directory):
     if not os.path.exists(directory):
diff --git a/Mflux_Comfy/Mflux_Core.py b/Mflux_Comfy/Mflux_Core.py
index c1612f5..6156532 100644
--- a/Mflux_Comfy/Mflux_Core.py
+++ b/Mflux_Comfy/Mflux_Core.py
@@ -4,8 +4,6 @@ import os
 import numpy as np
 import torch
 from tqdm import tqdm
-from PIL import Image
-from PIL.PngImagePlugin import PngInfo
 import comfy.utils as utils
 import folder_paths
 
@@ -21,7 +19,6 @@ from mflux.latent_creator.latent_creator import LatentCreator
 from mflux.post_processing.stepwise_handler import StepwiseHandler
 from mflux.controlnet.controlnet_util import ControlnetUtil
 from mflux.models.vae.vae import VAE
-import folder_paths
 from .Mflux_Pro import MfluxControlNetPipeline
 
 flux_cache = {}
@@ -163,4 +160,4 @@ def generate_image(prompt, model, seed, width, height, steps, guidance, quantize
     if tensor_image.dim() == 3:
         tensor_image = tensor_image.unsqueeze(0)
 
-    return (tensor_image,)
+    return (tensor_image,)
\ No newline at end of file