anubhav-singh commited on
Commit
dffb3ac
·
verified ·
1 Parent(s): 464350c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,11 +3,11 @@ from transformers import AutoModel,AutoModelForCausalLM, AutoTokenizer, Stopping
3
  import gradio as gr
4
  from threading import Thread
5
 
6
- model = AutoModelForCausalLM.from_pretrained(
7
  "DuckyBlender/racist-phi3",
8
 
9
  torch_dtype=torch.float16,
10
- trust_remote_code=True,
11
 
12
  )
13
  tokenizer = AutoTokenizer.from_pretrained("DuckyBlender/racist-phi3")
 
3
  import gradio as gr
4
  from threading import Thread
5
 
6
+ model = AutoModel.from_pretrained(
7
  "DuckyBlender/racist-phi3",
8
 
9
  torch_dtype=torch.float16,
10
+ trust_remote_code=True
11
 
12
  )
13
  tokenizer = AutoTokenizer.from_pretrained("DuckyBlender/racist-phi3")