Clemylia commited on
Commit
69e5a19
·
verified ·
1 Parent(s): f428148

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from fastapi_poe import PoeBot, make_app
7
  from fastapi_poe.types import QueryRequest, PartialResponse
8
 
9
  # --- CONFIGURATION DU MODÈLE ---
10
- MODEL_ID = "Finisha-F-scratch/InutileGheya"
11
  print("--> Chargement du tokenizer...")
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
13
 
@@ -42,7 +42,7 @@ class CharlottePoeBot(PoeBot):
42
  with torch.no_grad():
43
  outputs = model.generate(
44
  **inputs,
45
- max_new_tokens=150,
46
  do_sample=True,
47
  temperature=0.7,
48
  pad_token_id=tokenizer.eos_token_id
 
7
  from fastapi_poe.types import QueryRequest, PartialResponse
8
 
9
  # --- CONFIGURATION DU MODÈLE ---
10
+ MODEL_ID = "Finisha-F-scratch/Nelya-neko"
11
  print("--> Chargement du tokenizer...")
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
13
 
 
42
  with torch.no_grad():
43
  outputs = model.generate(
44
  **inputs,
45
+ max_new_tokens=200,
46
  do_sample=True,
47
  temperature=0.7,
48
  pad_token_id=tokenizer.eos_token_id