Added with torch no grad

app.py

@@ -94,7 +94,8 @@ def get_logits(caption: str, imgs: List[Image.Image]) -> torch.Tensor:
     inputs["pixel_values"] = (
         inputs["pixel_values"].half() if device == "cuda" else inputs["pixel_values"]
     )
-    outputs = model(**inputs)
+    with torch.no_grad():
+        outputs = model(**inputs)
     logits_per_image = outputs.logits_per_image
 
     return logits_per_image