Hugging Face's logo

Spaces:
ashioyajotham
/
falcon_7b_coder
Runtime error

App Files Community
1
ashioyajotham commited on
Commit
1f1c57f
·
1 Parent(s): b7096cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
  fine_tuned_model = "ashioyajotham/results"
3
- model = AutoModelForCausalLM.from_pretrained(fine_tuned_model)
4
  # Load the tokenizer for the Falcon 7B model with remote code trust
5
  tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model, trust_remote_code=True)
6
 
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
  fine_tuned_model = "ashioyajotham/results"
3
+ model = AutoModelForCausalLM.from_pretrained(fine_tuned_model,trust_remote_code=True)
4
  # Load the tokenizer for the Falcon 7B model with remote code trust
5
  tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model, trust_remote_code=True)
6