Update app.py
Browse files
app.py
CHANGED
@@ -27,10 +27,7 @@ ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type='colbert')
|
|
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 |
-
|
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}")
|
|
|
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 |
+
return response
|
|
|
|
|
|
|
31 |
|
32 |
def test_rag_reranking(query, ranker):
|
33 |
print(f"\n馃攳 Pregunta recibida: {query}")
|