Stevross commited on
Commit
78228e9
Β·
1 Parent(s): dfd2529

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from transformers import pipeline
5
  generate_text = pipeline(
6
  model="Stevross/Astrid-7B-Assistant-CPU",
7
  task="text-generation",
 
8
  device=0 # Use GPU if available, otherwise CPU
9
  )
10
 
 
5
  generate_text = pipeline(
6
  model="Stevross/Astrid-7B-Assistant-CPU",
7
  task="text-generation",
8
+ trust_remote_code=True,
9
  device=0 # Use GPU if available, otherwise CPU
10
  )
11