Update README.md

README.md

@@ -45,7 +45,9 @@ url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 texts = [["a photo of a cat", "a photo of a dog"]]
 inputs = processor(text=texts, images=image, return_tensors="pt")
-outputs = model(**inputs)
+
+with torch.no_grad():
+  outputs = model(**inputs)
 
 # Target image sizes (height, width) to rescale box predictions [batch_size, 2]
 target_sizes = torch.Tensor([image.size[::-1]])