Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from stablepy import Model_Diffusers
 
-from StableGR import (search_civitai, download_civitai, select_civitai_item, add_civitai_item, get_civitai_tag, select_civitai_all_item,
+from StableGR import (generate_image, search_civitai, download_civitai, select_civitai_item, add_civitai_item, get_civitai_tag, select_civitai_all_item,
                            update_civitai_selection, update_civitai_checkbox, from_civitai_checkbox,
                            CIVITAI_TYPE, CIVITAI_BASEMODEL, CIVITAI_SORT, CIVITAI_PERIOD, CIVITAI_FILETYPE, download_file)
 
@@ -16,25 +16,6 @@ css = """
 
 
 
-# Define the function to generate images
-def generate_image(model_id, prompt, lora_A, num_steps, guidance_scale, sampler, img_width, img_height):
-    model = Model_Diffusers(
-        base_model_id=model_id,
-        task_name='txt2img',
-    )
-
-    image, info_image = model(
-        prompt=prompt,
-        lora_A=lora_A,
-        num_steps=num_steps,
-        guidance_scale=guidance_scale,
-        sampler=sampler,
-        img_width=img_width,
-        img_height=img_height,
-    )
-    return image[0]
-
-
 with gr.Blocks(fill_width=True, css=css) as demo:
     with gr.Column():
         gr.Markdown("# StableGR", elem_classes="title")