omg typing

src/main.py

@@ -36,7 +36,7 @@ OUTPUT_DIR = "magellan-ai/brand-classifier-stage"
 print('logging in')
 login(token=HF_ACCESS_TOKEN)
 print('logged in')
-
+"""
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
@@ -46,7 +46,7 @@ print('setting device')
 device = torch.device("cuda")
 print(device)
 print('set device')
-
+"""
 print('setting up Repository')
 #TODO: Change based on whether input dir is stage to output to prod or other way 
 repo = hf.Repository(local_dir = REPOSITORY_PATH,
@@ -68,9 +68,9 @@ async def home():
 async def post_webhook(
 		payload: WebhookPayload,
 		task_queue: BackgroundTasks,
-		repo: repo,
-		REPOSITORY_PATH: REPOSITORY_PATH,
-		OUTPUT_DIR: OUTPUT_DIR,
+		repo = repo,
+		REPOSITORY_PATH: str = REPOSITORY_PATH,
+		OUTPUT_DIR: str = OUTPUT_DIR,
 		x_webhook_secret:  Optional[str] = Header(default=None),
 	):
 	print('received webhook:')

staging/brand-classifier-stage

@@ -0,0 +1 @@
+Subproject commit ce7a7c6ec3cc85c4695d2835623dff0fec95a2ff