AyoubChLin commited on
Commit
aa204f5
1 Parent(s): 31001c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -16,8 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3.5-mini-instruct")
16
  pipe = pipeline(
17
  "text-generation",
18
  model=model,
19
- tokenizer=tokenizer,
20
- device="cpu"
21
  )
22
 
23
  # Generation arguments
 
16
  pipe = pipeline(
17
  "text-generation",
18
  model=model,
19
+ tokenizer=tokenizer
 
20
  )
21
 
22
  # Generation arguments