Update app.py

app.py

@@ -4,14 +4,17 @@ from huggingface_hub import InferenceClient
 client = InferenceClient("bigscience/bloom")
 
 def generate_text(prompt, max_tokens, temperature, top_p):
-    response = client.chat_completion(
-        [{"role": "user", "content": prompt}],
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    )
-    return response
+    try:
+        response = client.chat_completion(
+            [{"role": "user", "content": prompt}],
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        )
+        return response
+    except Exception as e:
+        return str(e)
 
 demo = gr.Interface(
     fn=generate_text,