Rohit1412 commited on
Commit
16025b7
·
verified ·
1 Parent(s): 3a1d8c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def generate_response(query, pdf_file=None):
44
  full_input = query
45
 
46
  # Define the model and endpoint for the RAG model.
47
- model_id = "facebook/rag-token-nq"
48
  api_url = f"https://api-inference.huggingface.co/models/{model_id}"
49
 
50
  payload = {"inputs": full_input}
 
44
  full_input = query
45
 
46
  # Define the model and endpoint for the RAG model.
47
+ model_id = "google/flan-t5-large"
48
  api_url = f"https://api-inference.huggingface.co/models/{model_id}"
49
 
50
  payload = {"inputs": full_input}