Update app.py

Add models list filter for ahx only

app.py

@@ -123,8 +123,11 @@ def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, tok
   text_encoder.get_input_embeddings().weight.data[token_id] = embeds
   return token
 
+
+ahx_model_list = [model for model in models_list if "ahx" in model.modelId]
+
 print("Setting up the public library")
-for model in models_list:
+for model in ahx_model_list:
   model_content = {}
   model_id = model.modelId
   model_content["id"] = model_id