Update app.py

app.py

@@ -106,8 +106,8 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
     selected_info_2 = "Select a LoRA 2"
     lora_scale_1 = 1.15
     lora_scale_2 = 1.15
-    lora_image_1 = "/path/to/default/image.png"  # Provide a default image path
-    lora_image_2 = "/path/to/default/image.png"  # Provide a default image path
+    lora_image_1 = "/valid/path/to/default/image.png"  # Updated to a valid image path
+    lora_image_2 = "/valid/path/to/default/image.png"  # Updated to a valid image path
     if len(selected_indices) >= 1:
         lora1 = loras_state[selected_indices[0]]
         selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
@@ -188,7 +188,7 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
                 if repo.endswith(".safetensors") and repo.startswith("http"):
                     repo = download_file(repo)
                 new_item = {
-                    "image": image if image else "/home/user/app/custom.png",
+                    "image": image if image else "/valid/path/to/default/image.png",  # Updated to a valid image path
                     "title": title,
                     "repo": repo,
                     "weights": path,
@@ -285,7 +285,8 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
     with calculateDuration("Generating image"):
         # Generate image
         for img in pipe(prompt=prompt_mash, num_inference_steps=steps, guidance_scale=cfg_scale, width=width, height=height, generator=generator):
-            yield img
+            if img is not None:  # Check for NoneType before yielding
+                yield img
 
 @spaces.GPU(duration=75)
 def run_lora(prompt, selected_indices, loras_state):
@@ -394,7 +395,7 @@ def update_history(new_image, history):
         history = []
     history.insert(0, new_image)
     return history
-
+    
 css = '''
 #gen_btn{height: 100%}
 #title{text-align: center}