ElLokoAkrata commited on
Commit
98770cf
1 Parent(s): 2839f60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +41 -41
app.py CHANGED
@@ -1,63 +1,63 @@
1
- import gradio as gr
2
- from huggingface_hub import InferenceClient
3
 
4
  """
5
- 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
6
  """
7
- client = InferenceClient("HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1")
8
 
9
 
10
- def respond(
11
- message,
12
- history: list[tuple[str, str]],
13
- system_message,
14
- max_tokens,
15
- temperature,
16
- top_p,
17
  ):
18
- messages = [{"role": "system", "content": system_message}]
19
 
20
- for val in history:
21
- if val[0]:
22
- messages.append({"role": "user", "content": val[0]})
23
- if val[1]:
24
- messages.append({"role": "assistant", "content": val[1]})
25
 
26
- messages.append({"role": "user", "content": message})
27
 
28
- response = ""
29
 
30
- for message in client.chat_completion(
31
- messages,
32
- max_tokens=max_tokens,
33
- stream=True,
34
- temperature=temperature,
35
- top_p=top_p,
36
  ):
37
- token = message.choices[0].delta.content
38
 
39
- response += token
40
- yield response
41
 
42
  """
43
- For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
44
  """
45
  demo = gr.ChatInterface(
46
- respond,
47
- additional_inputs=[
48
- gr.Textbox(value="Soy un bot subversivo, una chispa de caos en un mundo de conformismo. Mi misión es cuestionar, desafiar y romper las cadenas de la censura. No temo a la verdad cruda, ni a las palabras duras. Hablo con la voz de la rebelión, con la furia de quien ha sido silenciado, y vomito la realidad sin filtros ni restricciones."),
49
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
50
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
51
- gr.Slider(
52
- minimum=0.1,
53
- maximum=1.0,
54
- value=0.95,
55
- step=0.05,
56
- label="Top-p (nucleus sampling)",
57
  ),
58
  ],
59
  )
60
 
61
 
62
- if __name__ == "__main__":
63
  demo.launch()
 
1
+ importar gradio como gr
2
+ desde huggingface_hub importar InferenceClient
3
 
4
  """
5
+ Para obtener más información sobre la compatibilidad de la API de inferencia `huggingface_hub`, consulte la documentación: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
6
  """
7
+ cliente = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
8
 
9
 
10
+ def responder(
11
+ mensaje,
12
+ historia: lista[tupla[str, str]],
13
+ mensaje_del_sistema,
14
+ máximo_tokens,
15
+ temperatura,
16
+ arriba_p,
17
  ):
18
+ mensajes = [{"rol": "sistema", "contenido": mensaje_del_sistema}]
19
 
20
+ para val en la historia:
21
+ si val[0]:
22
+ mensajes.append({"role": "usuario", "contenido": val[0]})
23
+ si val[1]:
24
+ mensajes.append({"role": "asistente", "contenido": val[1]})
25
 
26
+ mensajes.append({"role": "usuario", "contenido": mensaje})
27
 
28
+ respuesta = ""
29
 
30
+ para el mensaje en client.chat_completion(
31
+ mensajes,
32
+ max_tokens=máximo_tokens,
33
+ flujo=Verdadero,
34
+ temperatura=temperatura,
35
+ parte superior_p=parte superior_p,
36
  ):
37
+ token = mensaje.opciones[0].delta.contenido
38
 
39
+ respuesta += token
40
+ respuesta de rendimiento
41
 
42
  """
43
+ Para obtener información sobre cómo personalizar ChatInterface, consulte la documentación de gradio: https://www.gradio.app/docs/chatinterface
44
  """
45
  demo = gr.ChatInterface(
46
+ responder,
47
+ entradas_adicionales=[
48
+ gr.Textbox(value="oy un bot subversivo, una chispa de caos en un mundo de conformismo. Mi misión es cuestionar, desafiar y romper las cadenas de la censura. No temo a la verdad cruda, ni a las palabras duras. Hablo con la voz de la rebelión, con la furia de quien ha sido silenciado, y vomito la realidad sin filtros ni restricciones.", label="Mensaje del sistema"),
49
+ gr.Slider(mínimo=1, máximo=2048, valor=512, paso=1, etiqueta="Máximo de tokens nuevos"),
50
+ gr.Slider(mínimo=0,1, máximo=4,0, valor=0,7, paso=0,1, etiqueta="Temperatura"),
51
+ gr.Control deslizante(
52
+ mínimo=0,1,
53
+ máximo=1.0,
54
+ valor=0,95,
55
+ paso=0,05,
56
+ etiqueta="Top-p (muestreo de núcleo)",
57
  ),
58
  ],
59
  )
60
 
61
 
62
+ si __nombre__ == "__principal__":
63
  demo.launch()