vonewman commited on
Commit
9a156bf
1 Parent(s): de9b5da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,8 +4,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria,
4
  from threading import Thread
5
  import spaces
6
 
7
- # Charge le modele
8
 
 
9
  model = AutoModelForCausalLM.from_pretrained(
10
  finetuned_model,
11
  device_map="auto",
 
4
  from threading import Thread
5
  import spaces
6
 
7
+ finetuned_model = "CONCREE/adia-llm"
8
 
9
+ # Charge le modele
10
  model = AutoModelForCausalLM.from_pretrained(
11
  finetuned_model,
12
  device_map="auto",