Ryan/VideoDetect
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixes and new ai model

Browse Source
This commit is contained in:
Ryan Shpeherd
2026-09-25 22:40:24 -04:00
parent 8f2834775b
commit 589b775cda
2 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
aiworker/run.pl
+8 -6
View File
@@ -6,13 +6,15 @@ use JSON;
use Data::Dumper;
my $seconds = $ENV{'SECONDS_PER_FRAME'} || 30;
my $ollama = $ENV{'OLLAMA_URL'} || die "OLLAMA_URL environment variable not set\n";
my $api = $ENV{'API_BASE_URL'} || die "API_BASE_URL environment variable not set\n";
my $ollama = $ENV{'OLLAMA_URL'} || 'http://10.0.1.7:11434';
my $api = $ENV{'API_BASE_URL'} || 'http://10.0.1.7:3000';
my $task_count = $ENV{'TASK_COUNT'} || 1;
my $sensitivity = 20;
my $resolution = "672:672";
my $model = "hf.co/Jommarn/UNSEEN_Gemma_4_26B_NSFW-GGUF:Q8_0";
chdir("/tmp") || die "Unable to change directory to /tmp: $!\n";
my $ua = LWP::UserAgent->new;
$ua->timeout(60);
@@ -56,16 +58,16 @@ sub CheckPath {
$data=undef;
my $ref = {
'model' => 'srizon/pixie:latest',
'prompt' => "Are there any african american people in this photo, whether they are shown completely or not? Please answer 'yes' or 'no' only.",
'model' => $model,
'prompt' => "Are there any african american men in this photo, whether they are shown completely or not? Judge based on skin tone, accounting for shadows. Answer yes for very dark skin.Please answer 'yes' or 'no' only.",
'images' => [$base64],
};
my $url = "$ollama/api/generate";
my $res = GetRest($url, $ref);
print "\t$res\n";
print "$file\t$res\n";
$total++;
if($res=~/yes/i) {
if($res=~/yes$/i) {
$yes++;
}
unlink($file);