Spaces:
Running
Running
Update main.py
Browse files
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(
|
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])}
|