RodriiS commited on
Commit
c0bc251
·
verified ·
1 Parent(s): 3a186cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import pipeline
2
  import gradio as gr
3
 
4
  # Carrega um modelo de linguagem
5
- chat = pipeline("text-generation", model="google/flan-t5-small")
6
 
7
  def responder(msg):
8
  resposta = chat(msg, max_length=100, pad_token_id=50256)[0]['generated_text']
 
2
  import gradio as gr
3
 
4
  # Carrega um modelo de linguagem
5
+ chat = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
6
 
7
  def responder(msg):
8
  resposta = chat(msg, max_length=100, pad_token_id=50256)[0]['generated_text']