regarding error icon popping.

#86
by Bhaskar2611 - opened

I am using this model by downloading files manually and the chatbot interface directly by using the code from gradio but in this interface additionally, there are additional inputs and i am facing an issue with the error pop up after a response. how can i get rid of that?

image.png
is it because of my app.py if yes can anybody provide the correct app.py :) ??

Not used it in a while as i moved on quite quickly to others, but the last i heard gradio only connects to a back end API. So it would depend on if your code has some sort of back end running.

If im wrong and its not the case now, or if i misunderstood in the past, someone will correct me.

import gradio as gr
from huggingface_hub import InferenceClient

"""
For more information on huggingface_hub Inference API support, please check the docs: https://huggingface.co./docs/huggingface_hub/v0.22.2/en/guides/inference
"""
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")

def respond(
message,
history: list[tuple[str, str]],
system_message,
max_tokens,
temperature,
top_p,
):
messages = [{"role": "system", "content": system_message}]

for val in history:
    if val[0]:
        messages.append({"role": "user", "content": val[0]})
    if val[1]:
        messages.append({"role": "assistant", "content": val[1]})

messages.append({"role": "user", "content": message})

response = ""

for message in client.chat_completion(
    messages,
    max_tokens=max_tokens,
    stream=True,
    temperature=temperature,
    top_p=top_p,
):
    token = message.choices[0].delta.content

    response += token
    yield response

"""
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
"""
demo = gr.ChatInterface(
respond,
additional_inputs=[
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
gr.Slider(
minimum=0.1,
maximum=1.0,
value=0.95,
step=0.05,
label="Top-p (nucleus sampling)",
),
],
)

if name == "main":
demo.launch()

The above code is my app.py, which is for the traditional Gradio chatbot interface. Do I need to add any error handling cases or do I need to customize it further??

Sign up or log in to comment