Upload inference.py

inference.py

@@ -32,7 +32,7 @@ allowed_tags.append("safe")
 image_path = "/path/to/your/image.jpg"
 start = time.time()
 img = Image.open(image_path).convert('RGB')
-tensor = transform(img).unsqueeze(0).to("cuda")  # transform and add batch dimension
+tensor = transform(img).unsqueeze(0).to("cuda")
 
 with torch.no_grad():
     out = model(tensor)