GDrive exception handling (again)

SortAnon · Jul 23, 2021 · ccbd69c · ccbd69c
1 parent 1a47e6b
commit ccbd69c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/controllable_talknet.py b/controllable_talknet.py
@@ -722,7 +722,7 @@ def download_model(model, custom_model):
             drive_id = custom_model
         else:
             drive_id = model
-        if drive_id == "":
+        if drive_id == "" or drive_id is None:
             return ("Missing Drive ID", None, None)
         if not os.path.exists(os.path.join(UPLOAD_DIRECTORY, "models")):
             os.mkdir(os.path.join(UPLOAD_DIRECTORY, "models"))