hexanovapixel commited on
Commit
3b5f01c
·
verified ·
1 Parent(s): 4fbccab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,14 +1,15 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- chatbot = pipeline("text-generation", model="distilgpt2")
 
5
 
6
  def respond(message, history):
7
  if not message:
8
  return "Pregúntame algo 😊"
9
 
10
- prompt = "Responde en español de forma clara: " + message
11
- response = chatbot(prompt, max_length=100, do_sample=True)
12
 
13
  return response[0]["generated_text"]
14
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Modelo ligero y más coherente
5
+ chatbot = pipeline("text-generation", model="mrm8488/t5-base-spanish-summarization")
6
 
7
  def respond(message, history):
8
  if not message:
9
  return "Pregúntame algo 😊"
10
 
11
+ prompt = "Responde en español de forma clara y amigable: " + message
12
+ response = chatbot(prompt, max_length=100, do_sample=False)
13
 
14
  return response[0]["generated_text"]
15