aolko commited on
Commit
55ff40c
1 Parent(s): 3446156

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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")