Update StableGR.py

StableGR.py

@@ -45,6 +45,24 @@ def uniq_urls(s):
     return to_urls(list_uniq(parse_urls(s) + parse_repos(s)))
 
 
+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]
+
+
 
 def download_file(dl_url, civitai_key, progress=gr.Progress(track_tqdm=True)):
     download_dir = TEMP_DIR