model pathing

card_generator.py

@@ -23,7 +23,7 @@ test_item = {'Name': 'Pustulent Raspberry', 'Type': 'Fruit', 'Value': '1 cp', 'P
 # Function that takes in an image url and a dictionary and uses the values to print onto a card.
 def paste_image_and_resize(base_image,sticker_path, x_position, y_position,img_width, img_height, purchased_item_key = None):
     
-    # Check for if item has a Rarity string that is a in the dictionary of sticket paths
+    # Check for if item has a Rarity string that is in the dictionary of sticker paths
     if purchased_item_key:
         if sticker_path[purchased_item_key]:
             sticker_path = sticker_path[purchased_item_key]

img2img.py

@@ -9,11 +9,11 @@ from PIL import Image
 pipe = None
 start_time = time.time()
 torch.backends.cuda.matmul.allow_tf32 = True
-model_path = ("./models/stable-diffusion/card-generator-v1.safetensors")
-lora_path = "./models/stable-diffusion/Loras/blank-card-template-5.safetensors"
-detail_lora_path = "./models/stable-diffusion/Loras/add-detail-xl.safetensors"
-mimic_lora_path = "./models/stable-diffusion/Loras/EnvyMimicXL01.safetensors"
-temp_image_path = "./image_temp/"
+model_path = ("models/stable-diffusion/card-generator-v1.safetensors")
+lora_path = "models/stable-diffusion/Loras/blank-card-template-5.safetensors"
+detail_lora_path = "models/stable-diffusion/Loras/add-detail-xl.safetensors"
+mimic_lora_path = "models/stable-diffusion/Loras/EnvyMimicXL01.safetensors"
+temp_image_path = "image_temp/"
 card_pre_prompt = " blank magic card,high resolution, detailed intricate high quality border, textbox, high quality detailed magnum opus drawing of a "
 negative_prompts = "text, words, numbers, letters"
 image_list = []

main.py

@@ -137,35 +137,36 @@ with gr.Blocks() as demo:
             <h2><b>First:</b> Build a Card Template</h2>
                 </div>""")
     with gr.Row():
+        with gr.Column():
 
-    # Template Gallery instructions
-        gr.HTML(""" <div id="inner"> <header>
-                <h3>1. Click a border from the 'Card Template Gallery'</h3> 
-                </div>""")
-        
-    border_gallery = gr.Gallery(label = "Card Template Gallery", 
-                                    scale = 2,
-                                    value = useri.index_image_paths("Drakosfire/CardGenerator", "seed_images/card_templates"),
-                                    show_label = True,
-                                    columns = [3], rows = [3],
-                                    object_fit = "contain",
-                                    height = "auto",
-                                    elem_id = "Template Gallery")
-    
-    gr.HTML(""" <div id="inner"> <header>
-                <h3>2. Click a image from the Seed Image Gallery</h3><br>
-                </div>""")
-    
-    border_gallery.select(assign_img_path, outputs = selected_border_image)
-    seed_image_gallery = gr.Gallery(label= " Image Seed Gallery",
-                                    scale = 2,
-                                    value = useri.index_image_paths("Drakosfire/CardGenerator", "seed_images/item_seeds"),
-                                    show_label = True,
-                                    columns = [3], rows = [3],
-                                    object_fit = "contain",
-                                    height = "auto",
-                                    elem_id = "Template Gallery",
-                                    interactive=True)                 
+        # Template Gallery instructions
+            gr.HTML(""" <div id="inner"> <header>
+                    <h3>1. Click a border from the 'Card Template Gallery'</h3> 
+                    </div>""")
+            
+            border_gallery = gr.Gallery(label = "Card Template Gallery", 
+                                            scale = 2,
+                                            value = useri.index_image_paths("Drakosfire/CardGenerator", "seed_images/card_templates"),
+                                            show_label = True,
+                                            columns = [3], rows = [3],
+                                            object_fit = "contain",
+                                            height = "auto",
+                                            elem_id = "Template Gallery")
+        with gr.Column():
+            gr.HTML(""" <div id="inner"> <header>
+                        <h3>2. Click a image from the Seed Image Gallery</h3><br>
+                        </div>""")
+            
+            border_gallery.select(assign_img_path, outputs = selected_border_image)
+            seed_image_gallery = gr.Gallery(label= " Image Seed Gallery",
+                                            scale = 2,
+                                            value = useri.index_image_paths("Drakosfire/CardGenerator", "seed_images/item_seeds"),
+                                            show_label = True,
+                                            columns = [3], rows = [3],
+                                            object_fit = "contain",
+                                            height = "auto",
+                                            elem_id = "Template Gallery",
+                                            interactive=True)                 
 
     gr.HTML(""" <div id="inner"> <header><h4> -Or- Upload your own seed image, by dropping it into the 'Generated Template Gallery' </h4><br>
                 <h3>3. Click 'Generate Card Template'</h3><br>