alex-abb commited on
Commit
3b76ba2
1 Parent(s): 7f0947a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  import spaces
 
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
 
5
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
6
- model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
7
-
8
 
9
 
10
  # Optionnel : Fixer une graine aléatoire pour la reproductibilit
 
1
  import gradio as gr
2
  import spaces
3
+ # Load model directly
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
+ tokenizer = AutoTokenizer.from_pretrained("Sao10K/L3-8B-Stheno-v3.2")
7
+ model = AutoModelForCausalLM.from_pretrained("Sao10K/L3-8B-Stheno-v3.2")
 
8
 
9
 
10
  # Optionnel : Fixer une graine aléatoire pour la reproductibilit