bypass requirement for nsfw model
Some checks failed
release server / resolve build metadata (push) Successful in 6s
release server / build fluxer server (push) Successful in 3m36s
release server / create release (disabled for Gitea) (push) Has been skipped
release server / release summary (push) Successful in 5s
test cassandra-backup / Test latest Cassandra backup (push) Has been cancelled
Some checks failed
release server / resolve build metadata (push) Successful in 6s
release server / build fluxer server (push) Successful in 3m36s
release server / create release (disabled for Gitea) (push) Has been skipped
release server / release summary (push) Successful in 5s
test cassandra-backup / Test latest Cassandra backup (push) Has been cancelled
This commit is contained in:
@@ -49,6 +49,12 @@ export class NSFWDetectionService {
|
||||
}
|
||||
|
||||
async initialize(): Promise<void> {
|
||||
try {
|
||||
await fs.access(this.modelPath);
|
||||
} catch {
|
||||
this.session = null;
|
||||
return;
|
||||
}
|
||||
const modelBuffer = await fs.readFile(this.modelPath);
|
||||
this.session = await ort.InferenceSession.create(modelBuffer);
|
||||
}
|
||||
@@ -60,7 +66,7 @@ export class NSFWDetectionService {
|
||||
|
||||
async checkNSFWBuffer(buffer: Buffer): Promise<NSFWCheckResult> {
|
||||
if (!this.session) {
|
||||
throw new Error('NSFW Detection service not initialized');
|
||||
return {isNSFW: false, probability: 0};
|
||||
}
|
||||
|
||||
const processedImage = await this.preprocessImage(buffer);
|
||||
|
||||
Reference in New Issue
Block a user