pycharm中运行transformers模型的坑（anaconda prompt命令行运行正常，pycharm却报错）

pycharm中运行transformers模型的坑（anaconda prompt命令行运行正常，pycharm却报错）

1、transformer示例

from transformers import pipeline

nlp = pipeline("question-answering")

context = "Extractive Question Answering is the task of extracting an answer from a text given a question. An example of a question answering dataset is the SQuAD dataset, which is entirely based on that task. If you would like to fine-tune a model on a SQuAD task, you may leverage the `run_squad.py`."

print(nlp(question="What is extractive question answering?", context=context))
print(nlp(question="What is a good example of a question answering dataset?", context=context))

报错

但anaconda prompt运行正常

问题解决

修改后代码

from transformers import pipeline
if __name__ == '__main__':
    nlp = pipeline("question-answering")
    context = "Extractive Question Answering is the task of extracting an answer from a text given a question. An example of a question answering dataset is the SQuAD dataset, which is entirely based on that task. If you would like to fine-tune a model on a SQuAD task, you may leverage the `run_squad.py`."
    print(nlp(question="What is extractive question answering?", context=context))
    print(nlp(question="What is a good example of a question answering dataset?", context=context))

运行成功（注意缩进）