Update app.py

app.py

@@ -362,7 +362,7 @@ def add_custom_lora(custom_lora, selected_indices):
  if(repo.endswith(".safetensors") and repo.startswith("http")):
  repo = download_file(repo)
  new_item = {
- "image": image if image is not None else "",
+ "image": image,
  "title": title,
  "repo": repo,
  "weights": path,
@@ -390,22 +390,12 @@ def add_custom_lora(custom_lora, selected_indices):
  if len(selected_indices) >= 1:
  lora1 = loras[selected_indices[0]]
  selected_info_1 = f"### LoRA 1 Selected: {lora1['title']} ✨"
- lora_image_1 = lora1['image']
+ lora_image_1 = lora1['image'] if lora1['image'] else None
  if len(selected_indices) >= 2:
  lora2 = loras[selected_indices[1]]
  selected_info_2 = f"### LoRA 2 Selected: {lora2['title']} ✨"
- lora_image_2 = lora2['image']
+ lora_image_2 = lora2['image'] if lora2['image'] else None
  print("Chegou no final")
- print (
- gr.update(value=gallery_items),
- selected_info_1,
- selected_info_2,
- selected_indices,
- lora_scale_1,
- lora_scale_2,
- lora_image_1,
- lora_image_2
- )
  return (
  gr.update(value=gallery_items),
  selected_info_1,