AItool commited on
Commit
468788e
·
verified ·
1 Parent(s): ff38b89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,8 @@
3
  import gradio as gr
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
5
 
6
- model_id = "tiiuae/falcon-7b-instruct"
 
7
 
8
  # Load tokenizer and model with trust_remote_code
9
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
3
  import gradio as gr
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
5
 
6
+ model_id = "google/flan-t5-base" # ~250MB, loads fast
7
+
8
 
9
  # Load tokenizer and model with trust_remote_code
10
  tokenizer = AutoTokenizer.from_pretrained(model_id)