DHEIVER commited on
Commit
3a5bc19
·
verified ·
1 Parent(s): ca365ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -18
app.py CHANGED
@@ -41,57 +41,51 @@ class MistralChatbot:
41
  def launch_chat(self):
42
  additional_inputs=[
43
  gr.Slider(
44
- label="Temperatura",
45
  value=0.9,
46
  minimum=0.0,
47
  maximum=1.0,
48
  step=0.05,
49
  interactive=True,
50
- info="Valores mais altos produzem saídas mais diversas.",
51
  ),
52
  gr.Slider(
53
- label="Máximo de Novos Tokens",
54
  value=256,
55
  minimum=0,
56
  maximum=1048,
57
  step=64,
58
  interactive=True,
59
- info="O número máximo de novos tokens.",
60
  ),
61
  gr.Slider(
62
- label="Top-p (Amostragem de Núcleo)",
63
  value=0.90,
64
  minimum=0.0,
65
  maximum=1,
66
  step=0.05,
67
  interactive=True,
68
- info="Valores mais altos amostram mais tokens de baixa probabilidade.",
69
  ),
70
  gr.Slider(
71
- label="Penalidade de Repetição",
72
  value=1.2,
73
  minimum=1.0,
74
  maximum=2.0,
75
  step=0.05,
76
  interactive=True,
77
- info="Penalize tokens repetidos.",
78
  )
79
  ]
80
 
81
  gr.ChatInterface(
82
  fn=self.generate,
83
- inputs=["text", "text"] + ["slider"] * 4,
84
- outputs="text",
85
- title="Mistral 7B - Chatbot Médico",
86
- inputs_width="50%",
87
  additional_inputs=additional_inputs,
88
- theme="light", # Alteração para um tema mais claro
89
- show_typing_indicator=True,
90
- show_timestamps=True,
91
- show_multioutput=True
92
  ).launch(show_api=False)
93
 
94
- # Exemplo de uso:
95
  if __name__ == "__main__":
96
  chatbot = MistralChatbot()
97
- chatbot.launch_chat()
 
41
  def launch_chat(self):
42
  additional_inputs=[
43
  gr.Slider(
44
+ label="Temperature",
45
  value=0.9,
46
  minimum=0.0,
47
  maximum=1.0,
48
  step=0.05,
49
  interactive=True,
50
+ info="Higher values produce more diverse outputs",
51
  ),
52
  gr.Slider(
53
+ label="Max new tokens",
54
  value=256,
55
  minimum=0,
56
  maximum=1048,
57
  step=64,
58
  interactive=True,
59
+ info="The maximum numbers of new tokens",
60
  ),
61
  gr.Slider(
62
+ label="Top-p (nucleus sampling)",
63
  value=0.90,
64
  minimum=0.0,
65
  maximum=1,
66
  step=0.05,
67
  interactive=True,
68
+ info="Higher values sample more low-probability tokens",
69
  ),
70
  gr.Slider(
71
+ label="Repetition penalty",
72
  value=1.2,
73
  minimum=1.0,
74
  maximum=2.0,
75
  step=0.05,
76
  interactive=True,
77
+ info="Penalize repeated tokens",
78
  )
79
  ]
80
 
81
  gr.ChatInterface(
82
  fn=self.generate,
83
+ chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
 
 
 
84
  additional_inputs=additional_inputs,
85
+ title="Mistral 7B"
 
 
 
86
  ).launch(show_api=False)
87
 
88
+ # Example usage:
89
  if __name__ == "__main__":
90
  chatbot = MistralChatbot()
91
+ chatbot.launch_chat()