Show errors to user

app.py

@@ -3,7 +3,6 @@ import gradio as gr
 
 client = AsyncOpenAI()
 
-
 async def predict(message, history, model, temperature):
     history_openai_format = []
     for human, assistant in history:
@@ -11,19 +10,22 @@ async def predict(message, history, model, temperature):
         history_openai_format.append({"role": "assistant", "content": assistant})
     history_openai_format.append({"role": "user", "content": message})
 
-    response = await client.chat.completions.create(
-        model=model,
-        messages=history_openai_format,
-        temperature=temperature,
-        stream=True,
-    )
-
-    partial_message = ""
-    async for chunk in response:
-        if chunk.choices[0].delta.content is not None:
-            partial_message += chunk.choices[0].delta.content
-            yield partial_message
-
+    try:
+        response = await client.chat.completions.create(
+            model=model,
+            messages=history_openai_format,
+            temperature=temperature,
+            stream=True,
+        )
+
+        partial_message = ""
+        async for chunk in response:
+            if chunk.choices[0].delta.content is not None:
+                partial_message += chunk.choices[0].delta.content
+                yield partial_message
+
+    except Exception as e:
+         raise gr.Error(str(e))
 
 model_dropdown = gr.Dropdown(
     label="Model",