edi commited on
Commit
ab22e23
·
1 Parent(s): 6a577a7
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ compression_retriever = ContextualCompressionRetriever(
61
  # Configure FLAN-T5 (better task)
62
  # -----------------------------
63
  generator = pipeline(
64
- "text-generation", # 👈 Use this for T5 models
65
  model="google/flan-t5-base",
66
  max_new_tokens=1329, # shorter answers
67
  temperature=1,
 
61
  # Configure FLAN-T5 (better task)
62
  # -----------------------------
63
  generator = pipeline(
64
+ "text2text-generation", # 👈 Use this for T5 models
65
  model="google/flan-t5-base",
66
  max_new_tokens=1329, # shorter answers
67
  temperature=1,