<s>Human: 编写一个Python函数，以字符串作为输入，并返回文本中每个单词的个数。
“Hello，我是AtomGPT，我是一款语言模型。”
</s><s>Assistant: def count_words(text):
    words = text.split()
    word_counts = {}
    for word in words:
        if word in word_counts:
            word_counts[word] += 1
        else:
            word_counts[word] = 1
    return word_counts

text = "Hello，我是AtomGPT，我是一款语言模型。"
print(count_words(text))  # {'Hello，我是AtomGPT，我是一款语言模型。': 1}
</s>