From 95fc212dc9761a5ab945539ebec44e569007175c Mon Sep 17 00:00:00 2001
From: amirnd51 <amirnass@buffalo.edu>
Date: Wed, 24 Jul 2024 03:30:19 +0000
Subject: [PATCH] chore: Update gpu flag in predict function to use request
 value

---
 python_api/api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python_api/api.py b/python_api/api.py
index 0067e3b..4bdd740 100644
--- a/python_api/api.py
+++ b/python_api/api.py
@@ -319,7 +319,7 @@ async def predict(request: PredictRequest):
     architecture = request.architecture
     batch_size = request.batchSize
     desired_result_modality = request.desiredResultModality
-    gpu = True
+    gpu = request.gpu
     inputs = request.inputs
     # input_url = request.input_url
     model_id = request.model