ashutoshzade commited on
Commit
d2de40c
1 Parent(s): fc3624a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  from transformers import MistralForCausalLM
2
  from transformers import AutoTokenizer
3
 
 
 
 
4
  tokenizer = AutoTokenizer.from_pretrained('mistralai/mathstral-7B-v0.1')
5
 
6
  prompt = "What are the roots of unity?"
 
1
  from transformers import MistralForCausalLM
2
  from transformers import AutoTokenizer
3
 
4
+ from huggingface_hub import login
5
+ login()
6
+
7
  tokenizer = AutoTokenizer.from_pretrained('mistralai/mathstral-7B-v0.1')
8
 
9
  prompt = "What are the roots of unity?"