Hodely commited on
Commit
252f055
verified
1 Parent(s): 4b70789

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,19 +1,18 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # 馃 Modelo peque帽o, seguro y funcional sin GPU
5
- pipe = pipeline("text-generation", model="sshleifer/tiny-gpt2", device=-1)
6
 
7
  def responder(prompt):
8
- respuesta = pipe(prompt, max_new_tokens=60)[0]['generated_text']
9
  return respuesta
10
 
11
  demo = gr.Interface(
12
  fn=responder,
13
  inputs=gr.Textbox(label="Escribe tu mensaje"),
14
  outputs=gr.Textbox(label="Respuesta"),
15
- title="馃 AmInside 1.0 (Versi贸n segura)",
16
- description="Funciona gratis con ZeroGPU o CPU basic"
17
  )
18
 
19
  demo.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", device=-1)
 
5
 
6
  def responder(prompt):
7
+ respuesta = pipe(prompt, max_new_tokens=150, do_sample=True)[0]['generated_text']
8
  return respuesta
9
 
10
  demo = gr.Interface(
11
  fn=responder,
12
  inputs=gr.Textbox(label="Escribe tu mensaje"),
13
  outputs=gr.Textbox(label="Respuesta"),
14
+ title="馃 AmInside 1.0 (TinyLlama)",
15
+ description="Modelo TinyLlama en CPU, sin errores y sin gastar dinero 馃悜"
16
  )
17
 
18
  demo.launch()