placed return on cuda check for bot_comms()
Browse files
app.py
CHANGED
@@ -142,6 +142,7 @@ def bot_comms(input_text: str,
|
|
142 |
if input_text == "check cuda":
|
143 |
cuda_info = check_cuda()
|
144 |
yield cuda_info
|
|
|
145 |
|
146 |
if input_text == "switch to llama":
|
147 |
llm_mode = input_text
|
|
|
142 |
if input_text == "check cuda":
|
143 |
cuda_info = check_cuda()
|
144 |
yield cuda_info
|
145 |
+
return
|
146 |
|
147 |
if input_text == "switch to llama":
|
148 |
llm_mode = input_text
|