mirror of
				https://github.com/KevinMidboe/immich.git
				synced 2025-10-29 17:40:28 +00:00 
			
		
		
		
	chore(ml): updated dockerfile, added typing, packaging (#2642)
* updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose
This commit is contained in:
		
							
								
								
									
										64
									
								
								machine-learning/app/schemas.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										64
									
								
								machine-learning/app/schemas.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,64 @@ | ||||
| from pydantic import BaseModel | ||||
|  | ||||
|  | ||||
| def to_lower_camel(string: str) -> str: | ||||
|     tokens = [ | ||||
|         token.capitalize() if i > 0 else token | ||||
|         for i, token in enumerate(string.split("_")) | ||||
|     ] | ||||
|     return "".join(tokens) | ||||
|  | ||||
|  | ||||
| class VisionModelRequest(BaseModel): | ||||
|     image_path: str | ||||
|  | ||||
|     class Config: | ||||
|         alias_generator = to_lower_camel | ||||
|         allow_population_by_field_name = True | ||||
|  | ||||
|  | ||||
| class TextModelRequest(BaseModel): | ||||
|     text: str | ||||
|  | ||||
|  | ||||
| class TextResponse(BaseModel): | ||||
|     __root__: str | ||||
|  | ||||
|  | ||||
| class MessageResponse(BaseModel): | ||||
|     message: str | ||||
|  | ||||
|  | ||||
| class TagResponse(BaseModel): | ||||
|     __root__: list[str] | ||||
|  | ||||
|  | ||||
| class Embedding(BaseModel): | ||||
|     __root__: list[float] | ||||
|  | ||||
|  | ||||
| class EmbeddingResponse(BaseModel): | ||||
|     __root__: Embedding | ||||
|  | ||||
|  | ||||
| class BoundingBox(BaseModel): | ||||
|     x1: int | ||||
|     y1: int | ||||
|     x2: int | ||||
|     y2: int | ||||
|  | ||||
|  | ||||
| class Face(BaseModel): | ||||
|     image_width: int | ||||
|     image_height: int | ||||
|     bounding_box: BoundingBox | ||||
|     score: float | ||||
|     embedding: Embedding | ||||
|  | ||||
|     class Config: | ||||
|         alias_generator = to_lower_camel | ||||
|         allow_population_by_field_name = True | ||||
|  | ||||
|  | ||||
| class FaceResponse(BaseModel): | ||||
|     __root__: list[Face] | ||||
		Reference in New Issue
	
	Block a user