Merge pull request #31 from kms7530/feature-selective-load

[Feature] 할당하고 싶은 모델만 선택하여 `Bako` 서비스에 추가

Author: elan1537

models/vision_language.py

@@ -356,4 +356,4 @@ def infer_with_image(self, prompt: str, image: PILImage) -> str:
             image=image,
         )
 
-        return outputs
+        return outputs

parser/code_parser.py

@@ -83,7 +83,7 @@ def __parse_methods(self, class_node: ast.ClassDef) -> List[Dict]:
         for body_item in class_node.body:
             if isinstance(body_item, ast.FunctionDef):
                 method_info = self.__parse_function_signiture(body_item)
-                method_info["name"] = body_item.name     
+                method_info["name"] = body_item.name
 
                 for decorator in body_item.decorator_list:
                     if decorator_info := self.__extract_decorator(decorator):