tuan243 commited on
Commit
3946ae0
·
verified ·
1 Parent(s): ef01918

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -82,7 +82,7 @@ async def ai_personal_advice(request: AIRequest):
82
 
83
  Hãy tư vấn cách cải thiện hiệu suất làm việc dựa trên thông tin trên.
84
  """
85
- input_ids = tokenizer(combined_information, return_tensors="pt")
86
  response = ai_model.generate(**input_ids, max_new_tokens=500)
87
  # response = ai_model(prompt, max_length=200)
88
  return {"advice": tokenizer.decode(response[0])}
 
82
 
83
  Hãy tư vấn cách cải thiện hiệu suất làm việc dựa trên thông tin trên.
84
  """
85
+ input_ids = tokenizer(prompt, return_tensors="pt")
86
  response = ai_model.generate(**input_ids, max_new_tokens=500)
87
  # response = ai_model(prompt, max_length=200)
88
  return {"advice": tokenizer.decode(response[0])}