Spaces:

chansung
/

vid2persona

Paused

chansung commited on Mar 11, 2024

Commit

14a0dbb

verified ·

1 Parent(s): 2469b95

Update vid2persona/pipeline/llm.py

Files changed (1) hide show

vid2persona/pipeline/llm.py CHANGED Viewed

@@ -72,7 +72,7 @@ async def chat(
         for response in local_openllm.send_message(messages, model_id, max_input_token_length, parameters):
             yield response
     except ThreadPoolBuildError as e:
-        gr.Warning("Something went wrong. Switching to TGI remotely hosted model")
     finally:
         async for response in tgi_openllm.send_messages(messages, model_id, hf_token, parameters):
             yield response

         for response in local_openllm.send_message(messages, model_id, max_input_token_length, parameters):
             yield response
     except ThreadPoolBuildError as e:
+        gr.Warning(f"{e} ➡️ Switching to TGI remotely hosted model")
     finally:
         async for response in tgi_openllm.send_messages(messages, model_id, hf_token, parameters):
             yield response