joortif commited on
Commit
80fc204
verified
1 Parent(s): 7c37226

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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"Context: {context} Question: {query}"
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}")
 
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}")