AshenClock commited on
Commit
a2371d8
·
verified ·
1 Parent(s): f8f6117

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -331,14 +331,14 @@ async def call_hf_model(prompt: str, temperature: float = 0.5, max_tokens: int =
331
  logger.debug(f"PROMPT => {content_preview}")
332
  try:
333
  resp = client.text_generation(
334
- model=HF_MODEL,
335
- prompt, # Passa il prompt come argomento posizionale
336
  parameters={
337
  "temperature": temperature,
338
  "max_new_tokens": max_tokens,
339
  "top_p": 0.9
340
  }
341
- )
342
  # La struttura della risposta potrebbe variare; verifica il campo corretto
343
  raw = resp.get("generated_text", "")
344
  if not raw:
 
331
  logger.debug(f"PROMPT => {content_preview}")
332
  try:
333
  resp = client.text_generation(
334
+ HF_MODEL, # Argomento posizionale
335
+ prompt, # Argomento posizionale
336
  parameters={
337
  "temperature": temperature,
338
  "max_new_tokens": max_tokens,
339
  "top_p": 0.9
340
  }
341
+ )
342
  # La struttura della risposta potrebbe variare; verifica il campo corretto
343
  raw = resp.get("generated_text", "")
344
  if not raw: