Update app.py

app.py

@@ -33,12 +33,12 @@ app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_credentials=True,
 executor = ThreadPoolExecutor()
 
 # Determine the device (GPU or CPU)
-if torch.cuda.is_available():
-    device = "cuda"
-    logger.info("CUDA is available. Using GPU.")
-else:
-    device = "cpu"
-    logger.info("CUDA is not available. Using CPU.")
+# if torch.cuda.is_available():
+#     device = "cuda"
+#     logger.info("CUDA is available. Using GPU.")
+# else:
+device = "cpu"
+    # logger.info("CUDA is not available. Using CPU.")
 
 # Load model from Huggingface Hub
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")