From c718b1bdd043f0537c89629f0729797f04d5ea47 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lucca=20Baumg=C3=A4rtner?= <lucca.baumgaertner@fau.de>
Date: Mon, 5 Feb 2024 11:53:50 +0100
Subject: [PATCH] return to menu if invalid model name is given
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Lucca Baumgärtner <lucca.baumgaertner@fau.de>
---
 src/demo/demos.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/demo/demos.py b/src/demo/demos.py
index 3b0bdc7..2371f1d 100644
--- a/src/demo/demos.py
+++ b/src/demo/demos.py
@@ -387,6 +387,7 @@ def check_classification_task(string):
             log.info(f"Loaded the model from the local path sucessfully!")
     except:
         log.error("No model found with the given name!")
+        return
 
     if S3_bool:
         data_path = (