Update app.py

app.py

@@ -382,6 +382,13 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery, reque
         return (current_loras, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),gr.update(),
         )
 
+def process_custom_lora(custom_lora, request: gr.Request):
+    # Extract user token from request headers
+    user_token = request.headers.get("Authorization", "").replace("Bearer ", "")
+    if not user_token:
+        raise gr.Error("User is not logged in. Please log in to use this feature.")
+    return check_custom_model(custom_lora, token=user_token)
+
 def remove_custom_lora(selected_indices, current_loras, gallery):
     if current_loras:
         custom_lora_repo = current_loras[-1]['repo']
@@ -734,11 +741,13 @@ with gr.Blocks(theme=gr.themes.Soft(font=font), css=css, delete_cache=(128, 256)
         inputs=[custom_lora, selected_indices, loras_state, gallery],
         outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
     )
+
     remove_custom_lora_button.click(
         remove_custom_lora,
         inputs=[selected_indices, loras_state, gallery],
         outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4]
     )
+
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,