Upload 3 files

image_feature.py

@@ -146,14 +146,17 @@ def xiang(upload_image):
 
         image = upload_image.convert('RGB')
 
-        processor = AutoImageProcessor.from_pretrained('facebook/dinov2-giant')
-        model = AutoModel.from_pretrained('facebook/dinov2-giant')
+        # processor = AutoImageProcessor.from_pretrained('facebook/dinov2-giant')
+        # model = AutoModel.from_pretrained('facebook/dinov2-giant')
+
+        processor = AutoImageProcessor.from_pretrained('google/vit-large-patch16-224-in21k')
+        model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
 
         inputs = processor(images=image, return_tensors="pt").to(DEVICE)
         outputs = model(**inputs)
         # last_hidden_states = outputs.last_hidden_state
 
-        return outputs
+        return outputs.last_hidden_state
 
     except Exception as e:
         print(f"发生了一个错误: {e}")

requirements.txt

@@ -1,4 +1,5 @@
+numpy==1.26.3
+Pillow==10.2.0
+torch==2.1.2
 transformers==4.39.1
-torch
-gradio
-pillow
+opencv-python