sara.mesquita commited on
Commit
360f73a
·
1 Parent(s): b2ebfa6

changes to nvidia model

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. core/ai.py +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ Qualquer pessoa tira uma foto de um animal de rua pelo celular, o app detecta au
29
  | Secret | Descrição |
30
  |--------|-----------|
31
  | `HF_TOKEN` | Token HF com permissão de leitura (Settings → Tokens) |
32
- | `HF_MODEL` | Opcional. Padrão: `meta-llama/Llama-3.2-11B-Vision-Instruct` |
33
  | `MATCH_THRESHOLD` | Opcional. Threshold de similaridade. Padrão: `0.80` |
34
 
35
  > Os embeddings rodam com `sentence-transformers` **localmente no Space** (sem custo de API).
 
29
  | Secret | Descrição |
30
  |--------|-----------|
31
  | `HF_TOKEN` | Token HF com permissão de leitura (Settings → Tokens) |
32
+ | `HF_MODEL` | Opcional. Padrão: `nvidia/Llama-3.2-Nemotron-11B-Vision-Instruct` |
33
  | `MATCH_THRESHOLD` | Opcional. Threshold de similaridade. Padrão: `0.80` |
34
 
35
  > Os embeddings rodam com `sentence-transformers` **localmente no Space** (sem custo de API).
core/ai.py CHANGED
@@ -16,7 +16,7 @@ import numpy as np
16
 
17
  log = logging.getLogger(__name__)
18
 
19
- _DEFAULT_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct"
20
 
21
  PROMPT = (
22
  "You are analyzing a photo of a stray animal. "
 
16
 
17
  log = logging.getLogger(__name__)
18
 
19
+ _DEFAULT_MODEL = "nvidia/Llama-3.2-Nemotron-11B-Vision-Instruct"
20
 
21
  PROMPT = (
22
  "You are analyzing a photo of a stray animal. "