Spaces:

chansung
/

vid2persona

Paused

chansung commited on Mar 12

Commit

62d6792

•

1 Parent(s): 89fdef5

Update vid2persona/pipeline/llm.py

Files changed (1) hide show

vid2persona/pipeline/llm.py CHANGED Viewed

@@ -71,7 +71,7 @@ async def chat(
     try:
         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):

     try:
         for response in local_openllm.send_message(messages, model_id, max_input_token_length, parameters):
             yield response
+    except Exception 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):