BasqueLabs commited on
Commit
fca491e
verified
1 Parent(s): 473090c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  # Cargar el modelo Latxa
5
  def load_model():
6
- generator = pipeline("text-generation", model="HiTZ/latxa-7b-v1.1", device=-1) # Forzar uso de CPU
7
  return generator
8
 
9
  generator = load_model()
@@ -18,8 +18,8 @@ iface = gr.Interface(
18
  fn=generate_text,
19
  inputs="text",
20
  outputs="text",
21
- title="Generaci贸n de Texto con Latxa en Euskera",
22
- description="Escribe un texto en Euskera y el modelo Latxa generar谩 la continuaci贸n.",
23
  )
24
 
25
  if __name__ == "__main__":
 
3
 
4
  # Cargar el modelo Latxa
5
  def load_model():
6
+ generator = pipeline("text-generation", model="HiTZ/latxa-7b-v1.1") # Forzar uso de CPU
7
  return generator
8
 
9
  generator = load_model()
 
18
  fn=generate_text,
19
  inputs="text",
20
  outputs="text",
21
+ title="Latxa",
22
+ description="Idatzi testu bat euskaraz eta Latxa ereduak sortuko du jarraipena",
23
  )
24
 
25
  if __name__ == "__main__":