moooji
/

blip-vqa-capfilt-large

Inference Endpoints

Model card Files Files and versions Community

moooji commited on Feb 8, 2023

Commit

c6815f0

•

1 Parent(s): afc4db9

Update handler.py

Files changed (1) hide show

handler.py +1 -3

handler.py CHANGED Viewed

@@ -17,9 +17,7 @@ class EndpointHandler():
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
         inputs = self.processor(image, return_tensors="pt").to(device)
-        with torch.no_grad():
-            outputs = self.model(**inputs)
         return outputs

         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
         inputs = self.processor(image, return_tensors="pt").to(device)
+        outputs = self.model(**inputs)
         return outputs