SujathaL commited on
Commit
28341e6
·
verified ·
1 Parent(s): f6a0c41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
 
9
  # Load Mistral-7B Model for Generative Answers
10
- model_name = "mistralai/Mistral-7B-v0.1" # Smaller Mistral model
 
11
 
12
 
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
 
9
  # Load Mistral-7B Model for Generative Answers
10
+ model_name = "google/flan-t5-base" # Smallest alternative, works on free tier
11
+
12
 
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)