Update main.py

main.py

@@ -41,7 +41,7 @@ def generate(item: Item):
     model_inputs = encodeds.to(device)
 
 
-    generated_ids = model.generate(**model_inputs, item.max_new_tokens, do_sample=True, pad_token_id=tokenizer.eos_token_id, temperature=temperature)
+    generated_ids = model.generate(**model_inputs, item.max_new_tokens, do_sample=True, pad_token_id=tokenizer.eos_token_id, temperature=item.temperature)
     decoded = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
     return decoded