Functioned for Stef run

img2img.py

@@ -9,9 +9,9 @@ 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"
+model_path = ("/media/drakosfire/Shared/models/stable-diffusion/card-generator-v1/card-generator-v1.safetensors")
+lora_path = "/media/drakosfire/Shared/models/stable-diffusion/Loras/card-generator-v1/blank-card-template-5.safetensors"
+detail_lora_path = "./models/stable-diffusion/card-generator-v1/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 "

main.py

@@ -299,7 +299,7 @@ with gr.Blocks() as demo:
  
   
 if __name__ == '__main__':
-    demo.launch(server_name = "0.0.0.0", server_port = 8000, share = False, allowed_paths = ["/media/drakosfire/Shared/","/media/drakosfire/Shared/MerchantBot/card_templates"])
+    demo.launch(server_name = "0.0.0.0", server_port = 8000, share = True, allowed_paths = ["/media/drakosfire/Shared/","/media/drakosfire/Shared/MerchantBot/card_templates"])
         
 
 

user_input.py

@@ -21,7 +21,8 @@ def index_image_paths(repo_name,directory_path):
     files = []
     for content_file in contents:
         if content_file.type == "file":
-            files.append(content_file.download_url)  # Or content_file.path for just the path
+            media_url = content_file.download_url.replace("raw.githubusercontent.com", "media.githubusercontent.com/media")
+            files.append(media_url)  # Or content_file.path for just the path
     
     return files
 

utilities.py

@@ -37,15 +37,15 @@ def directory_contents(directory_path):
 
 # Delete a list of file 
 def delete_files(file_paths):
-     
-    for file_path in file_paths:
-        try:
-            os.remove(f"./image_temp/{file_path}")
-            print(f"Remove : ./image_temp/{file_path}")
-        except OSError as e:
-            print(f"Error: {file_path} : {e.strerror}")
-    file_paths.clear()
-    
+    if file_paths: 
+        for file_path in file_paths:
+            try:
+                os.remove(f"./image_temp/{file_path}")
+                print(f"Remove : ./image_temp/{file_path}")
+            except OSError as e:
+                print(f"Error: {file_path} : {e.strerror}")
+        file_paths.clear()
+        
 
 def open_image_from_url(image_url):
     response = requests.get(image_url)