Update app.py
Browse files
app.py
CHANGED
@@ -20,11 +20,11 @@ with open("Manual_de_nutrici贸n_clinica.pdf", "wb") as f:
|
|
20 |
f.write(response.content)
|
21 |
|
22 |
# Paso 2: Inicializar el modelo usando Hugging Face Pipeline
|
23 |
-
#
|
24 |
llm = HuggingFacePipeline.from_model_id(
|
25 |
-
model_id="
|
26 |
task="text-generation",
|
27 |
-
pipeline_kwargs={"temperature": 0, "top_k": 50, "top_p": 0.95, "max_length": 512}
|
28 |
)
|
29 |
chain = llm | StrOutputParser()
|
30 |
|
@@ -72,8 +72,8 @@ interface = gr.Interface(
|
|
72 |
inputs=gr.Textbox(label="Ingresa tu pregunta sobre nutrici贸n:"),
|
73 |
outputs=gr.Textbox(label="Respuesta:"),
|
74 |
title="Respuesta a Preguntas sobre Nutrici贸n",
|
75 |
-
description="Haz preguntas sobre nutrici贸n basadas en el Manual B谩sico de Nutrici贸n Cl铆nica y Diet茅tica"
|
76 |
)
|
77 |
|
78 |
# Lanzar la interfaz
|
79 |
-
interface.launch()
|
|
|
20 |
f.write(response.content)
|
21 |
|
22 |
# Paso 2: Inicializar el modelo usando Hugging Face Pipeline
|
23 |
+
# Usamos gpt2 en lugar de distilbert-base-uncased
|
24 |
llm = HuggingFacePipeline.from_model_id(
|
25 |
+
model_id="gpt2", # Cambiado a gpt2, que es compatible con AutoModelForCausalLM
|
26 |
task="text-generation",
|
27 |
+
pipeline_kwargs={"temperature": 0.7, "top_k": 50, "top_p": 0.95, "max_length": 512}
|
28 |
)
|
29 |
chain = llm | StrOutputParser()
|
30 |
|
|
|
72 |
inputs=gr.Textbox(label="Ingresa tu pregunta sobre nutrici贸n:"),
|
73 |
outputs=gr.Textbox(label="Respuesta:"),
|
74 |
title="Respuesta a Preguntas sobre Nutrici贸n",
|
75 |
+
description="Haz preguntas sobre nutrici贸n basadas en el Manual B谩sico de Nutrici贸n Cl铆nica y Diet茅tica."
|
76 |
)
|
77 |
|
78 |
# Lanzar la interfaz
|
79 |
+
interface.launch()
|