Update app.py

app.py

@@ -99,11 +99,11 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
     if selected_index in selected_indices:
         selected_indices.remove(selected_index)
     else:
-        if len(selected_indices) < 4:
+        if len(selected_indices) < 2:
             selected_indices.append(selected_index)
         else:
-            gr.Warning("You can select up to 4 LoRAs, remove one to select a new one.")
-            return gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), width, height, gr.update(), gr.update(), gr.update()
+            gr.Warning("You can select up to 2 LoRAs, remove one to select a new one.")
+            return gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), width, height, gr.update(), gr.update()
 
     selected_info_1 = "Select a LoRA 1"
     selected_info_2 = "Select a LoRA 2"
@@ -384,9 +384,9 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
         except Exception as e:
             print(e)
             gr.Warning(str(e))
-            return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
+            return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
     else:
-        return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
+        return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
 
 def remove_custom_lora(selected_indices, current_loras, gallery):
     if current_loras:
@@ -551,7 +551,7 @@ run_lora.zerogpu = True
 
 def get_huggingface_safetensors(link):
     split_link = link.split("/")
-    if len(split_link) == 4:
+    if len(split_link) == 2:
         model_card = ModelCard.load(link)
         base_model = model_card.data.get("base_model")
         print(f"Base model: {base_model}")