fix: remove broken models, add timm and huggingface_hub deps, add token login for gated repos

app.py

@@ -2,24 +2,26 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 import pandas as pd
+from huggingface_hub import login
 
 MIN_ACEPTABLE_SCORE = 0.1
 MAX_N_LABELS = 5
 
+HF_TOKEN = st.secrets.get("HF_TOKEN") if hasattr(st, "secrets") else None
+if HF_TOKEN:
+    login(token=HF_TOKEN)
+
 MODELS = [
             ("google/vit-base-patch16-224", "General Image Classification"),
             ("microsoft/resnet-50", "General Image Classification"),
             ("microsoft/resnet-18", "General Image Classification"),
             ("microsoft/resnet-34", "General Image Classification"),
             ("microsoft/resnet-101", "General Image Classification"),
-            ("microsoft/resnet-152", "General Image Classification"),
             ("microsoft/swin-tiny-patch4-window7-224", "General Image Classification"),
             ("microsoft/swinv2-base-patch4-window16-256", "General Image Classification"),
-            ("microsoft/beit-base-patch16-224-pt22k-ft22k", "General Image Classification"),
             ("facebook/convnext-large-224", "General Image Classification"),
             ("facebook/convnext-base-224-22k-1k", "General Image Classification"),
             ("facebook/convnext-tiny-224", "General Image Classification"),
-            ("nvidia/mit-b0", "General Image Classification"),
             ("timm/resnet50.a1_in1k", "General Image Classification"),
             ("timm/tf_efficientnetv2_s.in21k", "General Image Classification"),
             ("timm/convnext_tiny.fb_in22k", "General Image Classification"),
@@ -30,38 +32,24 @@ MODELS = [
             ("nateraw/vit-age-classifier", "Age Classification"),
 
             ("Falconsai/nsfw_image_detection", "NSFW Detection"),
-            ("LukeJacob2023/nsfw-image-detector", "NSFW Detection"),
             ("carbon225/vit-base-patch16-224-hentai", "NSFW Detection"),
             ("Marqo/nsfw-image-detection-384", "NSFW Detection"),
 
             ("cafeai/cafe_aesthetic", "Aesthetic/Art Classification"),
-            ("shadowlilac/aesthetic-shadow", "Aesthetic/Art Classification"),
-            ("pixai-labs/pixai-tagger-v0.9", "Aesthetic/Art Classification"),
 
             ("trpakov/vit-face-expression", "Face/Emotion Classification"),
             ("RickyIG/emotion_face_image_classification", "Face/Emotion Classification"),
             ("rizvandwiki/gender-classification", "Face/Emotion Classification"),
 
-            ("nateraw/food", "Food Classification"),
-            ("BinhQuocNguyen/food-recognition-model", "Food Classification"),
-
-            ("google/derm-foundation", "Medical/Dermatology"),
-            ("google/cxr-foundation", "Medical/Dermatology"),
             ("Anwarkh1/Skin_Cancer-Image_Classification", "Medical/Dermatology"),
 
             ("Ateeqq/ai-vs-human-image-detector", "AI vs Human Detection"),
             ("umm-maybe/AI-image-detector", "AI vs Human Detection"),
 
-            ("not-lain/deepfake", "Deepfake Detection"),
-
             ("DunnBC22/vit-base-patch16-224-in21k_Human_Activity_Recognition", "Human Activity Recognition"),
 
-            ("aalonso-developer/vit-base-patch16-224-in21k-clothing-classifier", "Clothing/Fashion"),
-
             ("andupets/real-estate-image-classification", "Real Estate"),
 
-            ("FatihC/swin-tiny-patch4-window7-224-finetuned-eurosat-watermark", "Satellite/Remote Sensing"),
-
             ("lamnt2008/car_brands_classification", "Car Classification"),
 
             ("docling-project/DocumentFigureClassifier-v2.5", "Document Classification"),

requirements.txt

@@ -1,3 +1,5 @@
 streamlit
 torch
-transformers
+transformers
+timm
+huggingface_hub