Update backend/main.py

backend/main.py

@@ -40,15 +40,6 @@ import socketio
 ###############################################
 # Configure logger
 
-# logging.basicConfig(filename="backend.log",
-#                     filemode='w',
-#                     format='%(asctime)s,%(msecs)d %(name)s %(levelname)s %(message)s',
-#                     datefmt='%H:%M:%S',
-#                     level=logging.DEBUG)
-
-# logger = logging.getLogger("socketio_server_pubsub")
-# gunicorn_logger.propagate = True
-
 gunicorn_error_logger = logging.getLogger("gunicorn.error")
 gunicorn_logger = logging.getLogger("gunicorn")
 uvicorn_access_logger = logging.getLogger("uvicorn.access")
@@ -143,8 +134,10 @@ static_files = {
 }
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large", cache_dir="/.cache")
+
 # PM - hardcoding temporarily as my GPU doesnt have enough vram
-model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large").to("cpu")
+# model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large").to("cpu")
+model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large").to(device)
 
 
 bytes_data = bytearray()