Update app.py

app.py

@@ -17,10 +17,27 @@ def image_mod(prompt: str, model: str, image_0: gr.Image, image_1: gr.Image) ->
         images[i] = pipeline(prompt).images[0]
     return images
 
+
+def get_text_to_image_models():
+    api = HfApi()
+
+    # Out app utilizers Diffusers and Safetensors
+    model_filter = ModelFilter(
+        library=["diffusers" , "safetensors"],
+        tags=["text-to-image"]
+    )
+
+    return api.list_models(filter=model_filter)
+
+
 def get_liked_models(username: str) -> list:
-    likes = list_liked_repos(username)
-    # filter if the models are diffusion models
-    return likes.models
+    user_liked_models = list_liked_repos(username)
+    text_to_image_models = get_text_to_image_models()
+    text_to_image_models_ids = [model.id for model in list(text_to_image_models)]
+
+    # find intersection between user liked models and models pulled from huggingface
+    user_liked_text_to_image_models = [i for i in user_liked_models.models if i in text_to_image_models_ids]
+    return user_liked_text_to_image_models
 
 
 def get_dropdown_for_huggingface(profile: gr.OAuthProfile):