mirror of
https://github.com/immich-app/immich
synced 2025-11-07 17:27:20 +00:00
18 lines
697 B
TypeScript
18 lines
697 B
TypeScript
|
import { MACHINE_LEARNING_URL } from '@app/common';
|
||
|
|
import { IMachineLearningRepository, MachineLearningInput } from '@app/domain';
|
||
|
|
import { Injectable } from '@nestjs/common';
|
||
|
|
import axios from 'axios';
|
||
|
|
|
||
|
|
const client = axios.create({ baseURL: MACHINE_LEARNING_URL });
|
||
|
|
|
||
|
|
@Injectable()
|
||
|
|
export class MachineLearningRepository implements IMachineLearningRepository {
|
||
|
|
tagImage(input: MachineLearningInput): Promise<string[]> {
|
||
|
|
return client.post<string[]>('/image-classifier/tag-image', input).then((res) => res.data);
|
||
|
|
}
|
||
|
|
|
||
|
|
detectObjects(input: MachineLearningInput): Promise<string[]> {
|
||
|
|
return client.post<string[]>('/object-detection/detect-object', input).then((res) => res.data);
|
||
|
|
}
|
||
|
|
}
|