edusc182 commited on
Commit
69c5ab1
·
verified ·
1 Parent(s): 99974d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # 1. Cargamos el modelo usando pipeline.
5
  # Reemplaza "TuUsuario/TuModelo" con la ruta real de tu modelo en Hugging Face.
6
  # Usamos "text-generation" asumiendo que tu modelo genera texto (como Llama, GPT, etc.)
7
- generador = pipeline("text-generation", model="TuUsuario/TuModelo")
8
 
9
  def predecir(pregunta):
10
  # 2. Pasamos la pregunta al modelo.
 
4
  # 1. Cargamos el modelo usando pipeline.
5
  # Reemplaza "TuUsuario/TuModelo" con la ruta real de tu modelo en Hugging Face.
6
  # Usamos "text-generation" asumiendo que tu modelo genera texto (como Llama, GPT, etc.)
7
+ generador = pipeline("text-generation", model="edusc182/Gemma_2B")
8
 
9
  def predecir(pregunta):
10
  # 2. Pasamos la pregunta al modelo.