asahi417 commited on
Commit
4dcbad1
1 Parent(s): efa4b4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,7 +33,8 @@ pipe = pipeline(
33
  batch_size=BATCH_SIZE,
34
  torch_dtype=torch_dtype,
35
  device=device,
36
- model_kwargs=model_kwargs
 
37
  )
38
 
39
 
 
33
  batch_size=BATCH_SIZE,
34
  torch_dtype=torch_dtype,
35
  device=device,
36
+ model_kwargs=model_kwargs,
37
+ trust_remote_code=True
38
  )
39
 
40