Hugging Face's logo

Spaces:
Chinnusamy
/
falcon11b
Runtime error

App Files Community
Chinnusamy commited on
Commit
6af617f
·
verified ·
1 Parent(s): 23b7957

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  # Load the model and tokenizer
5
- model_name = "tiiuae/falcon-11b"
 
6
  model = AutoModelForCausalLM.from_pretrained(model_name)
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
 
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  # Load the model and tokenizer
5
+ #model_name = "tiiuae/falcon-11b"
6
+ model_name = "databricks/dolly-v2-12b"
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9