Update app.py
Browse files
app.py
CHANGED
@@ -25,9 +25,12 @@ client = InferenceClient("google/flan-t5-base", token=os.getenv("HUGGINGFACEHUB_
|
|
25 |
ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type='colbert')
|
26 |
|
27 |
def generate_text(context, query):
|
28 |
-
inputs = f"
|
29 |
response = client.text_generation(prompt=inputs)
|
30 |
-
|
|
|
|
|
|
|
31 |
|
32 |
def test_rag_reranking(query, ranker):
|
33 |
print(f"\n馃攳 Pregunta recibida: {query}")
|
|
|
25 |
ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type='colbert')
|
26 |
|
27 |
def generate_text(context, query):
|
28 |
+
inputs = f"Contexto proporcionado: {context}\nPregunta: {query}\nResponde de manera clara y precisa sobre el tema relacionado con la constituci贸n espa帽ola."
|
29 |
response = client.text_generation(prompt=inputs)
|
30 |
+
if 'generated_text' in response:
|
31 |
+
return response['generated_text']
|
32 |
+
else:
|
33 |
+
return "No se pudo generar una respuesta adecuada."
|
34 |
|
35 |
def test_rag_reranking(query, ranker):
|
36 |
print(f"\n馃攳 Pregunta recibida: {query}")
|