Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,8 @@ from huggingface_hub import hf_hub_download
|
|
11 |
# Initialize models
|
12 |
anime_model_path = hf_hub_download("SmilingWolf/wd-convnext-tagger-v3", "model.onnx")
|
13 |
anime_model = ort.InferenceSession(anime_model_path)
|
14 |
-
photo_model = AutoModelForZeroShotImageClassification.from_pretrained("microsoft/Florence-2-base")
|
15 |
-
processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base")
|
16 |
|
17 |
# Load labels for the anime model
|
18 |
labels_path = hf_hub_download("SmilingWolf/wd-convnext-tagger-v3", "selected_tags.csv")
|
|
|
11 |
# Initialize models
|
12 |
anime_model_path = hf_hub_download("SmilingWolf/wd-convnext-tagger-v3", "model.onnx")
|
13 |
anime_model = ort.InferenceSession(anime_model_path)
|
14 |
+
photo_model = AutoModelForZeroShotImageClassification.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)
|
15 |
+
processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)
|
16 |
|
17 |
# Load labels for the anime model
|
18 |
labels_path = hf_hub_download("SmilingWolf/wd-convnext-tagger-v3", "selected_tags.csv")
|