fix path

webaverse · Oct 28, 2022 · 6a97541 · 6a97541
1 parent a58e6e2
commit 6a97541
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -98,7 +98,7 @@ async def upload_image(task: str = Form(), file: UploadFile = Form()):
 	feature_extraction_model.eval()
 	feature_extraction_model = feature_extraction_model.to(device)
 
-	image_text_matching_model = blip_decoder(pretrained=m'./models/model_base_retrieval_coco.pth', image_size=image_size, vit='base')
+	image_text_matching_model = blip_decoder(pretrained='./models/model_base_retrieval_coco.pth', image_size=image_size, vit='base')
 	image_text_matching_model.eval()
 	image_text_matching_model = image_text_matching_model.to(device)