Added some recent updates that were missing, and updated the filename with an if else.

In torch==1.5, the import of the API has changed. Although it does not interrupt the operation of the program, it seems to me to be an implicit error and may throw an exception in later versions.