Upload 2 files

image_feature.py

@@ -55,8 +55,8 @@ DEVICE = torch.device('cpu')
 # model = AutoModel.from_pretrained("google/vit-base-patch16-224-in21k").to(DEVICE)
 # processor = AutoImageProcessor.from_pretrained("chanhua/autotrain-izefx-v3qh0")
 # model = AutoModel.from_pretrained("chanhua/autotrain-izefx-v3qh0").to(DEVICE)
-processor = ViTImageProcessor.from_pretrained('google/vit-large-patch16-224-in21k')
-model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
+# processor = ViTImageProcessor.from_pretrained('google/vit-large-patch16-224-in21k')
+# model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
 # processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224-in21k')
 # model = ViTModel.from_pretrained('google/vit-base-patch16-224-in21k')
 
@@ -66,12 +66,13 @@ model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
 # pipe = pipeline(task="image-feature-extraction", model_name="google/vit-base-patch16-384", device=DEVICE, pool=True)
 # pipe = pipeline(task="image-feature-extraction", model_name="chanhua/autotrain-izefx-v3qh0", device=DEVICE, pool=True)
 # pipe = pipeline(task="image-feature-extraction", model_name="google/vit-base-patch16-224", device=DEVICE, pool=True, revision="29e7a1e183")
-pipe = pipeline(task="image-feature-extraction", model_name="google/vit-base-patch16-224-in21k", device=DEVICE, pool=True)
 
 
 # 推理
 def infer4(url1, url2):
  try:
+ pipe = pipeline(task="image-feature-extraction", model_name="google/vit-base-patch16-224-in21k", device=DEVICE, pool=True)
+
  print("进入推理")
  print("打开图片1")
  # image_real = Image.open(requests.get(url1, stream=True).raw).convert("RGB")
@@ -105,6 +106,8 @@ def infer4(url1, url2):
 
 # 推理
 def infer2(url):
+ processor = AutoImageProcessor.from_pretrained('google/vit-large-patch16-224-in21k')
+ model = ViTModel.from_pretrained('google/vit-large-patch16-224-in21k')
  # image_real = Image.open(requests.get(img_urls[0], stream=True).raw).convert("RGB")
  # image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
  image = Image.open(url).convert('RGB')