crmsolution commited on
Commit
2df5708
·
verified ·
1 Parent(s): f3bfdf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -1,21 +1,22 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Pipeline de geração de texto
5
  pipe = pipeline("text-generation", model="gpt2")
6
 
7
  # Função principal do CRM Doctor
8
  def chat(user_input):
9
- output = pipe(user_input, max_length=100)[0]["generated_text"]
10
- return output
11
 
12
- # Interface Gradio
13
  iface = gr.Interface(
14
  fn=chat,
15
  inputs=gr.Textbox(lines=3, placeholder="Descreva seus sintomas aqui..."),
16
  outputs="text",
17
  title="CRM Doctor",
18
- description="Seu assistente de pré-diagnóstico de saúde.\n\n🚨 *Este assistente não substitui uma consulta médica presencial.*"
19
  )
20
 
21
- iface.launch()
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Cria o pipeline usando o modelo GPT-2
5
  pipe = pipeline("text-generation", model="gpt2")
6
 
7
  # Função principal do CRM Doctor
8
  def chat(user_input):
9
+ result = pipe(user_input, max_length=100, do_sample=True)
10
+ return result[0]["generated_text"]
11
 
12
+ # Cria a interface Gradio
13
  iface = gr.Interface(
14
  fn=chat,
15
  inputs=gr.Textbox(lines=3, placeholder="Descreva seus sintomas aqui..."),
16
  outputs="text",
17
  title="CRM Doctor",
18
+ description="💙 Seu assistente de pré-diagnóstico.\n\n⚠️ *Esta IA não substitui consulta médica presencial.*"
19
  )
20
 
21
+ if __name__ == "__main__":
22
+ iface.launch()