Update main.py

main.py

@@ -17,6 +17,7 @@ app.add_middleware(
     allow_credentials=True,
 )
 
+
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32