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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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
- 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}")
 
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}")